當前位置：首頁 >

PyTorch view() 函数

發布時間：2025/4/5 81 豆豆

生活随笔收集整理的這篇文章主要介紹了 PyTorch view() 函数小編覺得挺不錯的,現在分享給大家,幫大家做個參考.

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() 函数的全部內容，希望文章能夠幫你解決所遇到的問題。

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