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)
print(z2)
print(z3)
print(z4)

1.3.1 补充：获得张量的形状

tensor.size()函数返回一个tensor.size类型的tuple，本质是tuple，支持一切tuple操作。

m=x.size()
print(m)
print(type(m))#本质上是一个tuple，支持一切元组操