torch.nn.modules.distance.CosineSimilarity Class Reference

Inheritance diagram for torch.nn.modules.distance.CosineSimilarity:

Public Member Functions
def	__init__ (self, dim=1, eps=1e-8)
def	forward (self, x1, x2)

Public Attributes
	dim
	eps

Detailed Description

Definition at line 48 of file distance.py.

---

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

• torch/nn/modules/distance.py