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)表示

>>阅读原文<<