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

>>阅读原文<<