pythorch中view函数

在 PyTorch 中，`view` 是一个张量（Tensor）的方法，用于改变张量的形状。它的作用类似于NumPy中的`reshape`方法。`view`的语法如下：

new_tensor = tensor.view(*shape)

其中，`tensor` 表示需要改变形状的张量，`shape` 是一个元组，表示新的形状。`new_tensor` 是改变形状后得到的新张量。需要注意的是，新张量的元素个数必须与原张量的元素个数相同。

`view`方法的一个常见用途是将一个高维张量展平成一个一维向量，例如：

import torch

# 创建一个大小为 (2, 3, 4) 的三维张量
x = torch.randn(2, 3, 4)
print(x.shape)  # 输出 (2, 3, 4)

# 将三维张量展平成一维向量
y = x.view(-1)
print(y.shape)  # 输出 (24)

上面的代码中，我们使用`view`方法将一个大小为`(2, 3, 4)`的三维张量`x`展平成了一个大小为`(24,)`的一维向量`y`。其中，`-1`表示自动推断该维度的大小。