Nikronic:

Could you please remove  autograd  tag from the question?
autograd  questions are about the autograd engine itself and it’s semantic not functions about tensors and manipulation of them.
Thanks


Yes, of course. Thanks for your reminder.