ott.neural.networks.layers.posdef.PositiveDense.rectifier_fn