Pytorch常用函數解析

torch模塊下的數學操作符 1. torch.numel() 返回一個tensor變量內所有元素個數，可以理解爲矩陣內元素的個數 2. torch.squeeze() 對於tensor變量進行維度壓縮，去除維數爲1的的維度。例如一矩陣維度爲A*1*B*C*1*D，通過squeeze()返回向量的維度爲A*B*C*D。squeeze(a)，表示將a的維數位1的維度刪掉，squeeze(a,N)表示

>>阅读原文<<