flatten() and unbind() in PyTorch

import torchmy_tensor = torch.tensor(2)torch.flatten(my_tensor)torch.flatten(my_tensor, 0, 0)torch.flatten(my_tensor, 0, -1)torch.flatten(my_tensor, -1, 0)torch.flatten(my_tensor, -1, -1)# tensor([2])my_tensor= torch.tensor([2, 7, 4]) # 1D tensortorch.flatten(my_tensor)torch.flatten(my_tensor, 0, 0)torch.flatten(my_tensor, 0, -1)torch.flatten(my_tensor, -1, 0)torch.flatten(my_tensor, -1, -1)# tensor([2, 7, 4])my_tensor = torch.tensor([[2, 7, 4], [8, 3, 2]]) # 2D tensortorch.flatten(my_tensor)torch.flatten(my_tensor, 0, 0)torch.flatten(my_tensor, 0, 1)torch.flatten(my_tensor, 0, -1)torch.flatten(my_tensor, -1, 0)torch.flatten(my_tensor, -1, -1)torch.flatten(my_tensor, -2, 1)torch.flatten(my_tensor, -2, -1)# tensor([2, 7, 4, 8, 3, 2])torch.flatten(my_tensor, 0, -2)torch.flatten(my_tensor, 1, 1)torch.flatten(my_tensor, 1, -1)torch.flatten(my_tensor, -1, 1)torch.flatten(my_tensor, -1, -1)torch.flatten(my_tensor, -2, 0)torch.flatten(my_tensor, -2, -2)# tensor([[2, 7, 4], [8, 3, 2]])my_tensor = torch.tensor([[[2, 7, 4], [8, 3, 2]], # 3D tensor                          [[5, 0, 8], [3, 6, 1]]])torch.flatten(my_tensor)torch.flatten(my_tensor, 0, 2)torch.flatten(my_tensor, 0, -1)torch.flatten(my_tensor, -3, -1)# tensor([2, 7, 4, 8, 3, 2, 5, 0, 8, 3, 6, 1])torch.flatten(my_tensor, 0, 0)torch.flatten(my_tensor, 0, -3)torch.flatten(my_tensor, 1, 1)torch.flatten(my_tensor, 1, -2)torch.flatten(my_tensor, 2, 2)torch.flatten(my_tensor, 2, -1)torch.flatten(my_tensor, -1, 2)torch.flatten(my_tensor, -1, -1)torch.flatten(my_tensor, -2, 1)torch.flatten(my_tensor, -2, -2)torch.flatten(my_tensor, -3, 0)torch.flatten(my_tensor, -3, -3)# tensor([[[2, 7, 4], [8, 3, 2]],#         [[5, 0, 8], [3, 6, 1]]])torch.flatten(my_tensor, 0, 1)torch.flatten(my_tensor, 0, -2)torch.flatten(my_tensor, -3, 1)torch.flatten(my_tensor, -3, 2)torch.flatten(my_tensor, -3, -2)# tensor([[2, 7, 4], [8, 3, 2], [5, 0, 8], [3, 6, 1]])torch.flatten(my_tensor, 1, 2)torch.flatten(my_tensor, 1, -1)torch.flatten(my_tensor, -2, 2)torch.flatten(my_tensor, -2, -1)# tensor([[2, 7, 4, 8, 3, 2],#         [5, 0, 8, 3, 6, 1]])

import torchmy_tensor= torch.tensor([2, 7, 4]) # 1D tensortorch.unbind(my_tensor)torch.unbind(my_tensor, 0)torch.unbind(my_tensor, -1)# (tensor(2), tensor(7), tensor(4))my_tensor = torch.tensor([[2, 7, 4], [8, 3, 2]]) # 2D tensortorch.unbind(my_tensor)torch.unbind(my_tensor, 0)torch.unbind(my_tensor, -2)# (tensor([2, 7, 4]), tensor([8, 3, 2]))torch.unbind(my_tensor, 1)torch.unbind(my_tensor, -1)# (tensor([2, 8]), tensor([7, 3]), tensor([4, 2]))my_tensor = torch.tensor([[[2, 7, 4], [8, 3, 2]], # 3D tensor                          [[5, 0, 8], [3, 6, 1]]])torch.unbind(my_tensor)torch.unbind(my_tensor, 0)torch.unbind(my_tensor, -3)# (tensor([[2, 7, 4], [8, 3, 2]]),#  tensor([[5, 0, 8], [3, 6, 1]]))torch.unbind(my_tensor, 1)torch.unbind(my_tensor, -2)# (tensor([[2, 7, 4], [5, 0, 8]]),#  tensor([[8, 3, 2], [3, 6, 1]]))torch.unbind(my_tensor, 2)torch.unbind(my_tensor, -1)# (tensor([[2, 8], [5, 3]]),#  tensor([[7, 3], [0, 6]]),#  tensor([[4, 2], [8, 1]]))