# ADD ONE DIMENSION: .unsqueeze(dim) my_tensor = torch.tensor([1,3,4]) # tensor([1,3,4]) my_tensor.unsqueeze(0) # tensor([[1,3,4]]) my_tensor.unsqueeze(1) # tensor([[1], # [3], # [4]])