pytorch programming

1. Convert torch.tensor to numpy:

a = torch.ones(5)

b = a.data.numpy()

2. Convert numpy to torch.tensor:

import numpy as np
a = np.ones(5)
b = torch.from_numpy(a)
np.add(a, 1, out=a)