3 ms·

Nice work! Does it support a dynamic graph of PyTorch?

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