Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
at::native::AdvancedIndex Struct Reference

Public Member Functions

 AdvancedIndex (const Tensor &src, TensorList indices)
 

Data Fields

Tensor src
 
std::vector< Tensorindices
 
DimVector indexed_sizes
 
DimVector indexed_strides
 
int64_t dims_before
 
int64_t dims_after
 

Detailed Description

Definition at line 290 of file Indexing.cpp.


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