athena.nll.ForwardNet.customized_loss

ForwardNet.customized_loss(inputs, mapped_inputs, gradients)[source]

Custom loss function.

Parameters

  • inputs (torch.Tensor) – DoubleTensor n_samples-by-n_params containing the points in the full input space.

  • mapped_inputs (torch.Tensor) – DoubleTensor n_samples-by-n_params containing the mapped points in the full input space. They are the result of the forward application.

  • gradients (torch.Tensor) – DoubleTensor n_samples-by-n_params containing the gradient samples wrt the input parameters.