范围在0-9之间的随机张量
时间: 2024-01-21 10:17:45 浏览: 64
PyTorch中张量的创建方法
下面是生成一个范围在0-9之间的随机张量的示例代码:
```python
import torch
# 生成形状为(3, 4)的随机整数张量
rand_tensor = torch.randint(low=0, high=10, size=(3, 4), dtype=torch.float)
print(rand_tensor)
```
输出结果类似于:
```
tensor([[ 6., 6., 3., 6.],
[ 8., 3., 7., 5.],
[ 9., 2., 2., 4.]])
```
其中 low=0 表示最小值为0,high=10 表示最大值为9(不包括10)。注意 dtype=torch.float 表示生成的张量数据类型为浮点型。如需整数型,请将 dtype 改为 torch.int。
阅读全文