Pytorch常用函数整理

1.torch.numel() 返回tensor变量内所有元素的个数，也可以简单理解为矩阵内yu元素的个数    例如，a的size为([64, 3, 7, 7])，那么a.numel() 返回值为64*3*7*7=9408 2.torch.squeeze() 将输入张量形状中的1去除并返回，如果输入是形如(Ax1xBx1xCx1xD)，那么输出形状就为(AxBxCxD) 3.torch.unsq

>>阅读原文<<