MallowDistanceLoss#

class torch_brain.nn.MallowDistanceLoss[source]#

Bases: torch_brain.nn.loss.Loss

forward(input, target, weights)[source]#

Compute Mallow distance loss.

Parameters:

input (Tensor) – The input tensor.
target (Tensor) – The target tensor.
weights (Tensor) – The weights tensor.

Return type: