torch array code example

Example 1: convert numpy to torch

torch.from_numpy(your_array)

Example 2: convert pytorch tensor to numpy

na = a.to('cpu').numpy()