pad_fwd

scarlet2.nn.pad_fwd(x, model_shape)

Zero-pads the input image to the model size

Parameters:

• x (jnp.array) – data to be padded

• model_shape (tuple) – shape of the prior model to be used

Returns:

• x (jnp.array) – data padded to same size as model_shape

• pad (tuple) – padding amount in every dimension