energyloss: Energy Loss Calculation

In DistributionIV: Distributional Instrumental Variable (DIV) Model

Energy Loss Calculation

Description

This function calculates the energy loss for given tensors. The loss is calculated as the mean of the L2 norms between 'x0' and 'x' and between 'x0' and 'xp', subtracted by half the mean of the L2 norm between 'x' and 'xp'.

Usage

energyloss(x0, x, xp, verbose = FALSE)

Arguments

x0	A tensor representing the target values.
x	A tensor representing the model's stochastic predictions.
xp	A tensor representing another draw of the model's stochastic predictions.
verbose	A boolean indicating whether to return prediction loss s1 = E(||x0-x||) and variance loss s2 = E(||x-xp||).

Value

A vector containing the calculated energy loss, 's1', and 's2', or a scalar representing the calculated energy loss.

DistributionIV documentation built on Sept. 9, 2025, 5:52 p.m.