<filename>forty/torch_utils.py
import numpy as np


def torch_to_numpy(x):
    return np.asarray(x.cpu().detach())

<filename>forty/torch_utils.py
import numpy as np


def torch_to_numpy(x):
    return np.asarray(x.cpu().detach())
