Pytorch笔记：张量操作

记录一些关于张量的操作，会持续更新

flatten()

用于展平张量的操作，目前我见到的有两个函数，分别是：

• torch.flatten()
• torch.Tensor.flatten()

二者其实是几乎一样的。以前者为例，函数原型为torch.flatten(input, start_dim=0,end_dim=-1)。其中start_dim是展平开始的维度，end_dim是展平结束的维度。例如，对于一个3x2x2的张量a，torch.flatten(a, start_dim=0,end_dim=-1)，会返回一个一维数组，torch.flatten(a, start_dim=0,end_dim=1)返回一个6x2的数组，torch.flatten(a, start_dim=1,end_dim=-1)返回一个3x4的数组。