pytorch torch.flatten()

Pytorch中torch.flatten()和torch.nn.Flatten()实例详解

 torch.flatten(x)等于torch.flatten(x，0)默认将张量拉成一维的向量，也就是说从第一维开始平坦化，torch.flatten(x，1)编程客栈代表从第二维开始平坦化。[详细]