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,

你可能感兴趣的:(pytorch,人工智能,python)