pytorch tensor创建

import torch
import numpy as np
 
print(torch.tensor([1,2,3]))
print(torch.tensor(np.arange(15).reshape(3,5)))
print(torch.empty([3,4]))
print(torch.ones([3,4]))
print(torch.zeros([3,4]))
#0-1之间的随机数
print(torch.rand([2,3]))
#3-10之间的随机整数
print(torch.randint(3,10,(2,2)))
#正态分布，均值为0，方差为1
print(torch.randn([3,4]))

转自：https://www.cnblogs.com/LiuXinyu12378/p/12292972.html