PyTorch | tensor的创建和基本运算

一、创建

1.1 基本的创建函数

x1=torch.empty(3,2)
x2=torch.rand(3,2)
x3=torch.zeros(3,2)
x4=torch.ones(3,2)
print(x1)
print(x2)
print(x3)
print(x4)

1.2 指定类型

上述的函数都可以用dtype参数指定类型

y2=torch.rand(3,2,dtype=torch.double)
y2

1.3 指定形状

可以用xx_like函数指定一个旧的tensor的形状

z1=torch.empty_like(x1)
z2=torch.rand_like(x1)
z3=torch.zeros_like(x1)
z4=torch.ones_like(x1)
print(z1)