pytorch中view函数用法

PyTorch中的`view()`函数用于改变张量的形状，也称为重塑或调整大小。它的参数是一个元组，表示新张量的形状。

例如，如果有一个形状为`(4, 3)`的张量`x`，我们可以使用`view()`函数将其重塑为`(6, 2)`的张量：

import torch

x = torch.randn(4, 3)
print(x.shape)  # 输出: torch.Size([4, 3])

y = x.view(6, 2)
print(y.shape)  # 输出: torch.Size([6, 2])

注意，`view()`函数返回一个新的张量，不会改变原始张量。如果重塑后的张量元素个数与原始张量不同，将会抛出异常。

此外，`view()`函数还可以使用`-1`作为一个维度的值，表示该维度的大小应该由其他维度推断而来。例如，我们可以将形状为`(4, 3)`的张量重塑为形状为`(2, -1)`的张量：

import torch

x = torch.randn(4, 3)
print(x.shape)  # 输出: torch.Size([4, 3])

y = x.view(2, -1)
print(y.shape)  # 输出: torch.Size([2, 6])