Class Gather

• Defined in File gather.h

Inheritance Relationships

Base Type

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

Class Documentation

class Gather : public mindspore::ops::PrimitiveC

Returns a slice of the input tensor based on the specified indices and axis. Refer to Python API mindspore.ops.Gather for more details.

Public Functions

inline Gather()

Constructor.

~Gather() = default

Destructor.

inline void Init()

Init. Refer to the parameters of Python API mindspore.ops.Gather for the inputs.