Skip to content

einops.asnumpy

Convert a tensor of an imperative framework (i.e. numpy/cupy/torch/jax/etc.) to numpy.ndarray

Parameters:

Name Type Description Default
tensor

tensor of any known imperative framework

required

Returns:

Type Description
Any

numpy.ndarray, converted to numpy

Source code in einops/einops.py
def asnumpy(tensor) -> np_ndarray:
    """
    Convert a tensor of an imperative framework (i.e. numpy/cupy/torch/jax/etc.) to `numpy.ndarray`

    Parameters:
        tensor: tensor of any known imperative framework

    Returns:
        `numpy.ndarray`, converted to numpy
    """
    return get_backend(tensor).to_numpy(tensor)