在PyTorch中,可以使用torch.rand()函数生成随机数。该函数返回一个张量,包含了从区间[0, 1)内均匀分布的随机数。例如:
import torch
# 生成一个大小为(2, 3)的随机数张量
random_tensor = torch.rand(2, 3)
print(random_tensor)
输出:
tensor([[0.7719, 0.0037, 0.5808],
[0.6445, 0.2133, 0.2590]])
另外,如果想要生成固定范围内的随机数,可以使用torch.randint()函数。该函数可以生成一个张量,包含了从给定范围内的离散均匀分布的随机整数。例如:
import torch
# 生成一个大小为(2, 3)、取值范围为[0, 10)的随机整数张量
random_int_tensor = torch.randint(0, 10, (2, 3))
print(random_int_tensor)
输出:
tensor([[5, 7, 9],
[1, 2, 0]])