pytorch view 函数介绍

view 是 PyTorch 中用于改变张量形状(tensor shape)的函数。与其他形状转换操作不同的是,view 并不改变张量的数据,而是返回一个新的张量,该张量与原始数据共享内存。

1. 基本用法

view 的作用是将一个张量重新排列成新的形状。它的基本语法是:

tensor.view(shape)
  • shape:新张量的形状,可以是整数或一个整数元组。shape 中的某一个维度可以是 -1,表示该维度的大小由张量的总元素个数和其他维度的大小自动推断。

2. 要求

为了使用 view 函数,张量必须是连续的(contiguous)。这意味着在内存中,张量的元素必须是按顺序存储的。如果张量不是连续的,可以先调用 contiguous() 方法使其连续。

3. 示例代码

import torch

# 创建一个形状为 (2, 3) 的张量
tensor = torch.tensor([[1, 2, 3], [4, 5, 6]])

# 将张量重塑为形状 (3, 2)
reshaped_tensor = tensor.view(3, 2)

print(reshaped_tensor)

解释:原始张量 tensor 形状为 (2,

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值