Class Pack
Defined in File pack.h
Inheritance Relationships
Base Type
public mindspore::ops::BaseOperator
Class Documentation
-
class Pack : public mindspore::ops::BaseOperator
Stacks a list of tensors in specified axis. Refer to Python API mindspore::ops::Stack for more details.
Public Functions
-
inline Pack()
Constructor.
-
void Init(const int64_t &axis = 0)
Init. Refer to the parameters of Python API mindspore::ops::Stack for the inputs.
-
void set_axis(const int64_t &axis)
Set axis.
-
int64_t get_axis() const
Get axis.
- Returns
axis.
-
inline Pack()