Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
caffe2.python.layers.pairwise_similarity.PairwiseSimilarity Class Reference
Inheritance diagram for caffe2.python.layers.pairwise_similarity.PairwiseSimilarity:

Public Member Functions

def __init__ (self, model, input_record, output_dim, pairwise_similarity_func='dot', name='pairwise_similarity', kwargs)
 
def add_ops (self, net)
 

Public Attributes

 pairwise_similarity_func
 
 indices_to_gather
 
 x_embeddings
 
 y_embeddings
 
 output_schema
 

Detailed Description

Definition at line 14 of file pairwise_similarity.py.


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