ott.solvers.linear.sinkhorn.SinkhornOutput.set_cost