Class EmbeddingLookup

• Defined in File embedding_lookup.h

Inheritance Relationships

Base Type

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

Class Documentation

class EmbeddingLookup : public mindspore::ops::PrimitiveC

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

Public Functions

inline EmbeddingLookup()

Constructor.

~EmbeddingLookup() = default

Destructor.

void Init(const bool setattr_flag = true)

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

void set_setattr_flag(const bool setattr_flag)

Set setattr_flag.

bool get_setattr_flag() const

Get setattr_flag.

Returns

setattr_flag.