Class TensorListGetItem
Defined in File tensor_list_get_item.h
Inheritance Relationships
Base Type
public mindspore::ops::BaseOperator
Class Documentation
-
class TensorListGetItem : public mindspore::ops::BaseOperator
TensorListGetItem defined TensorListGetItem operator prototype of lite.
Public Functions
-
inline TensorListGetItem()
Constructor.
-
void Init(const int64_t element_dtype)
Method to init the op’s attributes.
- Parameters
element_dtype – [in] Define the element data type in op;
-
void set_element_dtype(const int64_t element_dtype)
Method to set the op’s element_dtype attributes.
- Parameters
element_dtype – [in] Define the element data type in op;
-
int64_t get_element_dtype() const
Method to get the op’s element_dtype attributes.
-
inline TensorListGetItem()