常用到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 在列表末尾增加新的对象。