ott.solvers.linear.sinkhorn_lr.LRSinkhornOutput.set_cost