目录
- tensor的创建
- tensor的相关信息
- 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)
2.根据张量的形状创建
3.copy其他张量的形状创建
tensor的相关信息
- 1.数据类型
- 2.所在的设备
- 3.tensor的形状
tensor的运算
需要注意的是,在进行tensor计算时只有张量的形状一样(极其特殊时形状可以不同)且所在的设备一样才可以进行运算。
加,减,乘,除运算这里不再演示,就是按位进行加和减。
堆叠tensor
矩阵乘法