view ( *shape ) → Tensor
https://pytorch.org/docs/stable/tensors.html#torch.Tensor.view 1.3k
Returns a new tensor with the same data as the  self  tensor but of a different  shape .
torch. flatten ( input ,  start_dim=0 ,  end_dim=-1 ) → Tensor
https://pytorch.org/docs/stable/torch.html#torch.flatten 363
Flattens a contiguous range of dims in a tensor.