pytorch中與維度/變換相關的幾個函數

torch.size () 先說torch.size()函數，由於後面的方法都會用這個方法看到變換後的矩陣的維度 經過該方法，能夠查看當前Tensor的維度，用法也很簡單： >>>import torch >>>a = torch.Tensor([[[1, 2, 3], [4, 5, 6]]]) >>>a.size() torch.Size([1, 2, 3]) torch.view() 官方文檔

>>阅读原文<<