pytorch flatten()

 

 torch.flatten(input, start_dim, end_dim).

举例：一个tensor 3*2* 2

start_dim=1  output 3*4

start_dim=0 end_dim=1.    6*2

如果没有后面两个参数直接变为一维的