Class Attention
Defined in File attention.h
Inheritance Relationships
Base Type
public mindspore::ops::PrimitiveC
(Class PrimitiveC)
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.
-
inline Attention()