Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
torch.nn.modules.loss._Loss Class Reference
Inheritance diagram for torch.nn.modules.loss._Loss:
torch.nn.modules.loss._WeightedLoss torch.nn.modules.loss.BCEWithLogitsLoss torch.nn.modules.loss.CosineEmbeddingLoss torch.nn.modules.loss.CTCLoss torch.nn.modules.loss.HingeEmbeddingLoss torch.nn.modules.loss.KLDivLoss torch.nn.modules.loss.L1Loss torch.nn.modules.loss.MarginRankingLoss torch.nn.modules.loss.MSELoss torch.nn.modules.loss.MultiLabelMarginLoss torch.nn.modules.loss.PoissonNLLLoss torch.nn.modules.loss.SmoothL1Loss torch.nn.modules.loss.SoftMarginLoss torch.nn.modules.loss.TripletMarginLoss

Public Member Functions

def __init__ (self, size_average=None, reduce=None, reduction='mean')
 

Public Attributes

 reduction
 

Detailed Description

Definition at line 12 of file loss.py.


The documentation for this class was generated from the following file: