PyTorch中view()函数用法说明

首先,view( ) 是对 PyTorch 中的 Tensor 操作的,若非 Tensor 类型,可使用 data = torch.tensor(data)来进行转换。

(1) 作用:该函数返回一个有相同数据但不同维度大小的 Tensor。也就是说该函数的功能是改变矩阵维度,相当于 Numpy 中的 resize() 或者 Tensorflow 中的 reshape() 。

(2) 参数:view( *args )

python 复制代码
import torch

x = torch.randn(6, 6)
print(x.size())

y = x.view(36)
print(y.size())

z = x.view(-1, 9)  # -1表示该维度取决于其它维度大小,即(6*6)/ 9
print(z.size())

m = x.view(3, 3, 4) # 也可以变为更多维度
print(m.size())

输出:
torch.Size([6, 6])
torch.Size([36])
torch.Size([4, 9])
torch.Size([3, 3, 4])

特殊用法view(-1)

若需要转换维度为一维,有一种简单的方式,即将参数设置为 -1

python 复制代码
import torch

a = torch.Tensor([[1, 2, 3], [4, 5, 6],[7,8,9]]) # 定义一个 2*3 的 Tensor
a = a.view(-1)
print(a)

输出:
tensor([1., 2., 3., 4., 5., 6., 7., 8., 9.])
相关推荐
SpikeKing4 分钟前
Server - 使用 Docker 配置 PyTorch 研发环境
pytorch·docker·llm
丁先生qaq25 分钟前
热成像实例分割电力设备数据集(3类,838张)
人工智能·计算机视觉·目标跟踪·数据集
Eiceblue34 分钟前
Python读取PDF:文本、图片与文档属性
数据库·python·pdf
红衣小蛇妖1 小时前
神经网络-Day45
人工智能·深度学习·神经网络
weixin_527550401 小时前
初级程序员入门指南
javascript·python·算法
KKKlucifer1 小时前
当AI遇上防火墙:新一代智能安全解决方案全景解析
人工智能
程序员的世界你不懂1 小时前
Appium+python自动化(十)- 元素定位
python·appium·自动化
DisonTangor2 小时前
【小红书拥抱开源】小红书开源大规模混合专家模型——dots.llm1
人工智能·计算机视觉·开源·aigc
CryptoPP2 小时前
使用WebSocket实时获取印度股票数据源(无调用次数限制)实战
后端·python·websocket·网络协议·区块链
树叶@2 小时前
Python数据分析7
开发语言·python