ott.solvers.nn.icnn.ICNN.create_train_state#

ICNN.create_train_state(rng, optimizer, input)[source]#

Create initial TrainState.

Return type

TrainState

Parameters