标量表示
a = torch.tensor(1.2)
标量的shape和size都为空
向量表示
b= torch.tensor([1.0,2.])
c = torch.FloatTensor(3,3)
for i in [b,c]:
print(i.shape,i.size(),i.dim(),i.numel())
从numpy中创建tensor
import numpy as np
a = np.array([1,2,3])
b = torch.from_numpy(a)
print(b)
print("====")
a = np.ones([2,3])
b = torch.from_numpy(a)
print(b)
从list创建tensor
a = torch.tensor([1,2,3.])
print(a)
未初始化方法
a = torch.empty(2,3)
print(a)
print("======")
a = torch.FloatTensor(2,3,4)
print(a)
随机初值
a = torch.rand(2,3) #rand_like
print(a)
a = torch.randint(3,10,(3,4)) #最小值,最大值,size
print(a)
a = torch.rand(3,4)
print(a)
a = torch.full((3,4),7)
print(a)
a = torch.full((),7)
print(a)
a = torch.arange(2,5)
print(a)
a = torch.ones(2,3) #zeros eye
print(a)
a = torch.randperm(10) #随机打散
print(a)