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

Public Member Functions

def __init__ (self, model, input_record, name='merged')
 
def add_ops (self, net)
 

Public Attributes

 output_schema
 

Detailed Description

Merge multiple ID_LISTs into a single ID_LIST

Arguments:
    model: A layer model instance
    input_record: Tuple (Struct) of ID_LIST features to be
    merged

Returns:
    the merged ID_LIST feature

Definition at line 16 of file merge_id_lists.py.


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