Crate caffe2op_reservoir

source ·

Structs

  • ReservoirSamplingOp
    | Collect DATA tensor into RESERVOIR | of size num_to_collect. DATA is | assumed to be a batch. | | In case where ‘objects’ may be repeated | in data and you only want at most one instance | of each ‘object’ in the reservoir, OBJECT_ID | can be given for deduplication. If OBJECT_ID | is given, then you also need to supply | additional book-keeping tensors. | See input blob documentation for details. | | This operator is thread-safe. |