ott.solvers.nn.layers.PositiveDense.kernel_init

PositiveDense.kernel_init(shape, dtype=<class 'jax.numpy.float64'>)

Return type

Any

Parameters

• key (jax._src.prng.PRNGKeyArray) –

• shape (Sequence[Union[int, Any]]) –

• dtype (Any) –