torch.rand参数

torch.rand()函数是PyTorch中用于生成随机数的函数。它返回一个张量，该张量的元素是在区间[0, 1)内均匀分布的随机数。

torch.rand(size)函数可以接受一个参数size，用于指定生成随机数张量的形状。size可以是一个整数，表示生成一个一维随机数张量，也可以是一个元组，表示生成一个多维随机数张量。

下面是使用torch.rand()函数生成随机数张量的示例：

import torch

# 生成一个一维随机数张量
tensor1d = torch.rand(5)
print(tensor1d)

# 生成一个二维随机数张量
tensor2d = torch.rand((3, 4))
print(tensor2d)

输出示例：

tensor([0.2507, 0.8431, 0.8746, 0.8709, 0.6652])
tensor([[0.9806, 0.4118, 0.9329, 0.0989],
        [0.8177, 0.5338, 0.9707, 0.8792],
        [0.4424, 0.8966, 0.8913, 0.3802]])