pytorch
文章平均质量分 59
pytorch中常用代码函数分享
whitesilence
这个作者很懒,什么都没留下…
展开
-
torch tensor维度变换torch.reshape,torch.view,torch.repeat,torch.expand,torch.permute,torch.transpose
pytorch中可以改变维度的操作有torch.reshape,torch.view,torch.repeat,torch.expand,torch.permute,torch.transposeimport torch#torch.view和torch.reshape是一样的,torch.reshape/torch.view与input共用基础数据,改变其中一个,另一个也跟着改变t = torch.rand(4,4)print(t)b = t.view(2,8) #按行优先对原tensor进行原创 2021-08-10 18:08:54 · 2921 阅读 · 0 评论 -
pytorch 中的乘法*,@,torch.dot(), torch.matmul(), torch.mm(), torch.mul(), torch.bmm()
简介pytorch中实现乘法的操作有*,@,dot(),matmul(),mm(),mul(),bmm()*,@是两个运算符,他们分别映射到函数torch.mul和torch.matmul()运算符映射函数表https://docs.python.org/3/library/operator.html#mapping-operators-to-functions广播机制(摘自https://blog.csdn.net/MrR1ght/article/details/105660981 )在具体叨叨原创 2021-07-23 14:22:24 · 6444 阅读 · 4 评论 -
pytorch torch.gather 和 torch.scatter_
torch.gather(input, dim, index, *, sparse_grad=False, out=None) → Tensor依据dim指定的轴,取出input中index指定处的数据,输出与index相同shape的tensorinput和index必须有相同的维数,对于不是指定的dim维,index的维度==input的维度。对于在指定的dim维,index的维度<=input的维度import torch#当指定维度为0时,以列为单位,每列的值按照index中每列指定原创 2021-07-28 17:33:02 · 280 阅读 · 0 评论