torch.randint_like
时间: 2024-06-03 19:03:08 浏览: 249
torch.cuda.is_available()返回False解决方案
5星 · 资源好评率100%
`torch.randint_like`函数可以生成与输入张量形状相同的随机整数张量。可以指定生成整数的范围和数据类型。下面是一个使用`torch.randint_like`函数的例子:
```python
import torch
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
y = torch.randint_like(x, low=0, high=10, dtype=torch.int32)
print(y)
```
上述代码中,我们首先创建了一个形状为`(2, 3)`的张量`x`,然后使用`torch.randint_like`函数生成了一个与`x`形状相同的随机整数张量`y`,其中整数的范围是`[0, 10)`,数据类型是`torch.int32`。最后打印输出了生成的随机整数张量`y`。
阅读全文