介绍

在 PyTorch 中，view 是一个非常重要的函数，用于对 Tensor 进行形状变换。其中，view(-1) 是一种常见的用法，它的含义是将 Tensor 改变为一维，且该维度大小由程序自动推断。

语法

view() 函数的语法如下：

view(*shape) -> Tensor

其中，shape 表示变换后的形状，可以是一个 tuple 或多个整数。另外，view_as() 函数可以根据另一个 Tensor 的形状进行变换。

view(-1) 表示将 Tensor 变换为一维，且该维度大小由程序自动推断。

示例

以下示例展示了如何使用 view(-1) 对 Tensor 进行变换：

import torch

x = torch.rand(2, 3, 4)
print(x.size())  # 输出：torch.Size([2, 3, 4])

y = x.view(-1)
print(y.size())  # 输出：torch.Size([24])

总结

view(-1) 是一个非常方便的函数，可以在需要将 Tensor 转换为一维时，避免手动计算维度大小。需要注意的是，变换后的 Tensor 的元素个数需要与原 Tensor 的元素个数相同。除此之外，还需要注意维度的顺序是否正确。