Class TensorListStack

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.