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

Public Member Functions

def __init__ (self, margin=0., size_average=None, reduce=None, reduction='mean')
def forward (self, input1, input2, target)
- Public Member Functions inherited from torch.nn.modules.loss._Loss
def __init__ (self, size_average=None, reduce=None, reduction='mean')

Public Attributes

- Public Attributes inherited from torch.nn.modules.loss._Loss

Detailed Description

Definition at line 985 of file

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