squarenet.backends

Functions

from_backend(x)

Convert any supported array/tensor to a NumPy ndarray.

get_backend(x)

Return backend for a supported array.

to_backend(x[,Β backend])

Convert arrays between NumPy, Torch and JAX.
squarenet.backends.from_backend(x)[source]

Convert any supported array/tensor to a NumPy ndarray.

squarenet.backends.get_backend(x)[source]

Return backend for a supported array.

squarenet.backends.to_backend(x, backend='numpy')[source]

Convert arrays between NumPy, Torch and JAX.

Philosophy

  • If backend already match, return x unchanged.

  • Otherwise always convert through NumPy.