Loss# class torch_brain.nn.Loss[source]# Bases: torch.nn.modules.module.Module, abc.ABC Base class for losses. All losses should support an optional weights argument. abstract forward(input, target, weights=None)[source]# Abstract method for computing the loss. Return type: Tensor
