view() 函數

view(-1) 展平數組

import torch a = torch.Tensor([[1,2,3],[4,5,6]]) a.view(-1) >>> tensor([1., 2., 3., 4., 5., 6.]) a.view(-1).shape >>> torch.Size([6])

view(3,-1) 展平為3行

import torch a = torch.Tensor([[1,2,3],[4,5,6]]) a.view(3,-1) >>> tensor([[1., 2.],[3., 4.],[5., 6.]]) a.view(3,-1).shape >>> torch.Size([3, 2])

總結

以上是生活随笔為你收集整理的PyTorch view() 函数的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。