neuralop.losses.equation_losses.ICLoss

class neuralop.losses.equation_losses.ICLoss(loss=<function mse_loss>)[source]

Computes loss for initial value problems.

Extracts the initial condition and computes the loss between predicted and true initial conditions for all channels.

Expected input shape: (batch_size, channels, time_dim, *spatial_dims)

Methods

__call__(y_pred, y, **kwargs)

Expected input shape: (batch_size, channels, time_dim, *spatial_dims)