Class Pack

• Defined in File pack.h

Inheritance Relationships

Base Type

• public mindspore::ops::PrimitiveC (Class PrimitiveC)

Class Documentation

class Pack : public mindspore::ops::PrimitiveC

Stacks a list of tensors in specified axis. Refer to Python API mindspore::ops::Stack for more details.

Public Functions

inline Pack()

Constructor.

~Pack() = default

Destructor.

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.