对于3维矩阵,dim为-1时 与dim为2时的效果是一样的。dim为0时 从0维度,下图 是三维实例 图的目的是 可以由一个想象的空间。下面代码与上图关系不大>>>ab=torch.tensor([[[0,1,2,3],[1,2,3,4]],[[2,3,4,5],[4,5,6,7]],[[5,6,7,8],[6,7,8,9]]])>>>abtensor([[[0,1,2,3],[1,2,3,4]],[[2,3,4,5],[4,5,6,7]],[[5,6,7,8],[6,7,8,9]]])>>>ab.shape#ab的形状尺寸torch.Size([3,2,4])>>>ab.size()#ab的
1.torch.sumtorch.sum(imgs,dim=0)#按列求和torch.sum(imgs,dim=1) #按行求和imgs=torch.Tensor([iforiinzip(range(10),range(10))])print(imgs)s1=torch.sum(imgs,dim=0)#按列求和s2=torch.sum(imgs,dim=1)#按行求和print(s1)print(s2)2.torch.unsqueeze将每个图像张量的维度扩展,即在每个图像张量的最前面添加一个额外维度,以匹配模型的输入形状要求。 torch.unsqueeze(i,dim=0) #按dim=0
Tensor加法:a=torch.tensor([1,2,3])b=torch.tensor([4,5,6])c=a+bprint(c)#tensor([5,7,9])c=torch.add(a,b)print(c)#tensor([5,7,9])c=a.add(b)print(c)#tensor([5,7,9])Tensor减法:a=torch.tensor([1,2,3])b=torch.tensor([4,5,6])c=a-bprint(c)#tensor([-3,-3,-3])c=torch.sub(a,b)print(c)#tensor([-3,-3,-3])c=a.sub(b)pr