PyTorch生成随机张量的函数

1、torch.rand()

该函数用于生成一个张量，其中的元素是在区间[0, 1)内均匀分布的随机数

import torch


random_tensor = torch.rand(2, 3)
random_like_input = torch.rand_like(random_tensor)

2、torch.randn()

该函数生成一个张量，其中的元素是从标准正态分布（均值为0，标准差为1）中抽取的随机数

import torch


random_normal_tensor = torch.randn(2, 3)
random_normal_like_input = torch.randn_like(random_normal_tensor)

3、torch.randint()

该函数生成一个张量，其中的元素是在指定范围内的整数随机数

import torch


random_int_tensor = torch.randint(low=0, high=10, size=(2, 3))
random_int_like_input = torch.randint_like(random_int_tensor)

4、torch.bernoulli()

该函数生成生成服从伯努利分布的随机数

import torch


random_bernoulli_tensor = torch.bernoulli(0.5 * torch.ones(2, 3))

5、torch.normal()

该函数用于生成服从指定均值和标准差的正态分布的随机数张量

import torch


mean = torch.tensor([0.0, 1.0])
std = torch.tensor([1.0, 2.0])
random_normal_tensor = torch.normal(mean, std)