Class Attention

Inheritance Relationships

Base Type

Class Documentation

class Attention : public mindspore::ops::PrimitiveC

MultiHead-Attention op in MindIR.

Public Functions

inline Attention()

Constructor.

~Attention() override = default

Destructor.

inline void Init()

Initialize Attention op.