1.將數據切分

pytorch: tensor_data.view(N,T,C,V,W)? #例，將tensor數據拆分為N,T,C,V,W。允許設置-1，計算默認維度的值

numpy：numpy_data.reshape(N,T,C,V,W) #例，將numpy數據拆分為N,T,C,V,W。

2.改變數據維度

pytorch：tensor_data.permute(0,1,3,2,4)? #例，將2,3維度調換。

numpy：numpy_data.transpose(0,1,4,3,2) #例，將?2,4維度調換。

3.沿某一維度取平均

pytorch：tensor_data.mean(dim,keepdims = ?) #例，將dim維度取均值,取-1代表最后一個維度,keepdims表示是否保留此維度,取值為0或1

numpy：numpy_data.mean(dim,keepdims = ?) #

4.對數組求平方

pytorch：tensor_data.pow(2) #

numpy: numpy_data.square() #

5.沿某一維度相加

pytorch：tensor_data.sum(dim,keepdims = ?) #例，將dim維度求和,keepdims表示是否保留此維度,取值為0或1

numpy：numpy_data.sum(dim,keep_dims = ?)

6.求內積：不可用數據調用

pytorch：torch.matmul(a,b) # import torch

numpy: numpy.matmul(a,b) # import numpy

? ? ? ? ? ? ?numpy.dot(a,b) #常用于低維

7.矩陣相乘： a*b即可

8.去除維度值為1的維度

pytorch：torch_data.squeeze(dim) #將dim維度去除，注意只有當dim = 1時才能被去除，不帶任何參數時將所有維度為1的刪除，-1表示最后一個。-2表示倒數第二個.... 。

numpy：numpy_data.squeeze(dim) #同上。

9，添加維度為1的維度

pytorch：torch_data.unsqueeze(dim) #在dim處添加值為1的維度必須帶參數，-1表示最后一個位置。-2表示倒數第二個位置... 。

numpy：np.expand_dims(a,dim = ?) #這里不能用數據調用！！！

！！以上均展示的為數據直接調用，無特殊說明，均可以替換為np.xxxx(data,...) 或 torch.xxxx(data,....)

總結

以上是生活随笔為你收集整理的pytorch对张量的一些常用处理以及numpy对数组的一些常用处理的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。