How to create a tensor in pytorch