backend.utils

Utility functions for working with different backends.

To add support for a new backend, add a conversion function to the CONVERTERS list.

Kramer Harrison, 2024

Functions

is_torch_tensor(obj)	Checks if an object is a PyTorch tensor.
to_numpy(obj)	Converts input scalar or array to NumPy array, regardless of backend.
torch_to_numpy(obj)	Convert a torch Tensor to a NumPy array; raise TypeError otherwise.

is_torch_tensor(obj) → bool

Checks if an object is a PyTorch tensor.

Parameters:

obj – The object to check.

Returns:

True if the object is a PyTorch tensor, False otherwise.

Return type:

bool

to_numpy(obj: ScalarOrArrayT) → NDArray

Converts input scalar or array to NumPy array, regardless of backend.

torch_to_numpy(obj: Tensor) → NDArray

Convert a torch Tensor to a NumPy array; raise TypeError otherwise.