Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
torch.nn.modules.loss.TripletMarginLoss Class Reference
Inheritance diagram for torch.nn.modules.loss.TripletMarginLoss:
torch.nn.modules.loss._Loss

Public Member Functions

def __init__ (self, margin=1.0, p=2., eps=1e-6, swap=False, size_average=None, reduce=None, reduction='mean')
 
def forward (self, anchor, positive, negative)
 
- Public Member Functions inherited from torch.nn.modules.loss._Loss
def __init__ (self, size_average=None, reduce=None, reduction='mean')
 

Public Attributes

 margin
 
 p
 
 eps
 
 swap
 
- Public Attributes inherited from torch.nn.modules.loss._Loss
 reduction
 

Detailed Description

Definition at line 1145 of file loss.py.


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