PyTorch 深度学习实践
文章平均质量分 80
个人学习记录
奔跑的林小川
慎终如始,则无败事
展开
-
Pytorch(2):关于torch.cat()与torch.stack()
用法:torch.cat(): 用于连接两个相同大小的张量torch.stack(): 用于连接两个相同大小的张量,并扩展维度实例:import torcha = torch.tensor(torch.arange(10)).reshape(3, 3)b = torch.tensor(torch.arange(10, 100, 10)).reshape(3, 3)print(a)Out[7]: tensor([[1, 2, 3], [4, 5, 6],原创 2022-05-29 10:23:16 · 1355 阅读 · 1 评论 -
Pytorch (1): 张量的广播和运算
一. 主题:0. 理清Pytorch中广播的计算原理1. 记录Pytorch中较为特殊的计算方法2. 记录Pytorch矩阵计算方法二. 广播:当张量的维度shape不一致时,广播机制会自动调整张量维度使得计算可以顺利进行。实例1:张量t2.shape = (3, 4),张量t21.shape = (1, 4),其计算过程如下,将t21复制为shape=(3, 4),然后对应位置相加计算。计算结果:实例2:当两个张量的形状上有两个分量不同时,只要不同的分量仍然原创 2022-04-01 16:34:30 · 2184 阅读 · 0 评论