當(dāng)前位置：首頁 > 编程资源 > 编程问答 >内容正文

编程问答

PyTorch view() 函数

發(fā)布時間：2025/4/5 编程问答 67 豆豆

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

view(-1) 展平數(shù)組

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。