如何通过torch.flatten，将(197, 3, 1)降维成（197，）

可以通过以下代码将形状为(197, 3, 1)的张量降维为形状为(197,)的张量：

import torch

x = torch.randn(197, 3, 1)  # 创建一个形状为(197, 3, 1)的张量
x_flattened = torch.flatten(x)  # 将张量降维为形状为(197,)的张量
print(x_flattened.shape)  # 输出张量的形状

输出结果为：

torch.Size([197])

其中，`torch.flatten()`函数可以将任意形状的张量降维成一维张量，其参数`start_dim`指定了从哪一维开始展开，`end_dim`指定了展开到哪一维结束。若不指定任何参数，则默认展开所有维度。在本例中，由于只有最后一维的长度为1，因此使用默认参数即可将整个张量展开成一维张量。