Class TensorListStack
Defined in File tensor_list_stack.h
Inheritance Relationships
Base Type
public mindspore::ops::BaseOperator
Class Documentation
-
class TensorListStack : public mindspore::ops::BaseOperator
TensorListStack defined TensorListStack operator prototype of lite.
Public Functions
-
inline TensorListStack()
Constructor.
-
void Init(const int64_t num_elements, const int64_t element_dtype)
Method to init the op’s attributes.
- Parameters
num_elements – [in] Define the number elements in op;
element_dtype – [in] Define the element data type in op;
-
void set_num_elements(const int64_t num_elements)
Method to set the op’s num_elements attributes.
- Parameters
element_dtype – [in] Define the num_elements 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_num_elements() const
Method to get the op’s num_elements attributes.
-
int64_t get_element_dtype() const
Method to get the op’s element_dtype attributes.
-
inline TensorListStack()