pytorch生成随机数

在Pytorch中生成随机数可以使用torch.rand()函数。该函数会返回一个张量，张量中的每个元素都是在[0,1)之间的随机数。

示例代码：

import torch

# 生成一个形状为(2,3)的随机张量
x = torch.rand(2,3)
print(x)

输出：

tensor([[0.7918, 0.6094, 0.2744],
        [0.7663, 0.2186, 0.3341]])

除了torch.rand()函数，Pytorch还提供了其他生成随机数张量的函数，例如：

- torch.randn()：生成正态分布的随机数张量。
- torch.randint()：生成指定范围内的整数随机数张量。
- torch.randperm()：生成一个随机排列的整数序列。