Comment by apaszke

It supports numpy, and conversion from PyTorch tensors to numpy arrays is a matter of calling the numpy() method.