ott.geometry.low_rank.LRCGeometry.apply_square_cost

LRCGeometry.apply_square_cost(arr, axis=0)

Apply elementwise-square of cost matrix to array (vector or matrix).

Parameters:

• arr (Array) –

• axis (int) –

Return type:

Array