torch.nn.modules.loss.TripletMarginLoss Class Reference

Inheritance diagram for torch.nn.modules.loss.TripletMarginLoss:

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:

• torch/nn/modules/loss.py