python 迁移代码常用函数——修改数据至原代码所需要的格式,降低代码迁移难度

2021-06-03 08:03

阅读:682

标签:常用   去掉   span   matrix   squeeze   app   坐标   常用函数   依次   

常用到torch中的几个函数:

主要针对矩阵,在python中所有的矩阵都可以看作由0~n维的空间构成,类似于空间坐标。

  • transpose((x, y, z)) 转轴  这里的x,y,z其实指的是0,1,2维,正常顺序应该是0,1,2,但是当变为1,0,2时,即将1维转向0维。
  • view 改变矩阵维度,类似于reshape()函数,重新定义矩阵形状。当view中一个参数定为-1时,代表动态调整这个维度上的元素个数,从而保证元素总个数不变。
  • narrow(axis,start,length) 留下某维上的某些值,具体指的是留下第axis维上的第start到start+length这些元素。
  • repeat(matrix,number,axis) 重复matrix,number指重复次数,axis为None矩阵平铺后,元素依次重复number次;axis为0,增加行数;axis为1,增加列数。
  • unsqueeze(axis) 在第axis维上增加1维,通常就是增加[],axis=-2表示倒数第2维。
  • squeeze(axis) 在第axis维上减少1维,通常就是减少[]。只有该维为1维时,才能去掉。

针对list最常见的函数:

append 在列表末尾增加新的对象。

python 迁移代码常用函数——修改数据至原代码所需要的格式,降低代码迁移难度

标签:常用   去掉   span   matrix   squeeze   app   坐标   常用函数   依次   

原文地址:https://www.cnblogs.com/kemibyr/p/14678794.html


评论


亲,登录后才可以留言!