Tutorials:
Benchmarks:
Advanced Applications:
Public API: ott packages
References:
Apply elementwise-square of cost matrix to array (vector or matrix).
ndarray
arr (jax._src.numpy.ndarray.ndarray) –
axis (int) –