常用pytorch函数

unsqueeze(-1)在最后一个维度增加一个维度,[2,3] ->[2,3,1]

squeeze(2)在第二个维度减少一个维度(只有维度为1时才会真正被去掉) ,[2,1,3]->[2,3];如果是[2,2,3]在执行后还是[2,2,3]

permute(1,2,0) 交换维度,如[32,64,3]变为[64,3,32]

clip(0.,1.) 将数值限制在0.0到1.0之间,小于0.0的赋值为0.0,大于1.0的赋值为1.0 ,[-0.5,0.5,1.5]->[0.0,0.5,1.0]

detach 从计算图上分离下来,不更新

单下划线 _ :有时候单个独立下划线是用作一个名字,来表示某个变量是临时的或无关紧要的。

如也可以在拆分(unpacking)表达式中将单个下划线用作“不关心的”变量,以忽略特定的值。

car = ('red', 'auto', 12, 3812.4)    

 color, _, _, mileage = car     #来获取仅关心的颜色与里程数


版权声明:本文为weirdo_coder原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。