pad_back# scarlet2.nn.pad_back(x, pad)[source]# Removes the zero-padding from the input image Parameters: x (jnp.array) – padded data to same size as model_shape pad (tuple) – padding amount in every dimension Returns: x – data returned to it pre-pad shape Return type: jnp.array
