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 |
|
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)