TuckerTensorTrain.to_jax

t3toolbox.tucker_tensor_train.TuckerTensorTrain.to_jax()

def to_jax(self) -> 'TuckerTensorTrain':

Convert core arrays defining TuckerTensorTrain to Jax arrays.

Returns:

• TuckerTensorTrain – Tucker tensor train where tucker_cores and tt_cores are jax arrays.

• See Also

• ———

• TuckerTensorTrain.to_numpy()

Return type:

TuckerTensorTrain