PyTorch学习之torch.reshape函数

PyTorch学习之torch.reshape函数

一、什么是 torch.reshape

torch.reshape 函数用于将一个张量重新排列成指定的形状（shape）。其本质上是创建了一个新的张量，新的张量与原始张量共享相同的数据存储，但形状发生了变化。

二、torch.reshape 的语法

torch.reshape(input, shape)

• input：待调整形状的张量。
• shape：一个表示新形状的元组或列表。新形状的元素乘积必须等于原始张量的元素总数。

三、示例

下面通过几个简单的例子来演示 torch.reshape 的用法。

示例 1：基本用法

import torch

# 创建一个 1x8 的张量
x = torch.arange(8)
print("原始张量：")
print(x)

# 将 1x8 的张量重塑为 2x4 的张量，下面给两条语句的作用相同
# x_reshaped = torch.reshape(x, (2, 4))
x_reshaped