PyTorch Tensors 2 Numpy Adaptors¶
import torch
class TensorNumpyAdapter:
"""
Class for adapter interface between numpy array
type and Tensor objects in PyTorch.
"""
def to_tensor(self, x):
return torch.from_numpy(x).float()
def to_numpy(self, x):
return x.numpy()
Source: PyGlow Example