PyTorch入门之【tensor】

tensor的创建

1.手动创建

import torch
test1=torch.tensor([1,2,3])#一维时为向量
test2=torch.tensor([[1,2,3]])#二维时为矩阵
test3=torch.tensor([[[1,2,3]]])#三维及以上统称为tensor
print(test1)
print(test2)
print(test3)
1
2
3
4
5
6
7

2.根据张量的形状创建

3.copy其他张量的形状创建

tensor的相关信息

• 1.数据类型
• 2.所在的设备
• 3.tensor的形状

tensor的运算

需要注意的是，在进行tensor计算时只有张量的形状一样(极其特殊时形状可以不同)且所在的设备一样才可以进行运算。
加，减，乘，除运算这里不再演示，就是按位进行加和减。
堆叠tensor

矩阵乘法