Class Attention
Defined in File attention.h
Inheritance Relationships
Base Type
public mindspore::ops::BaseOperator
Class Documentation
-
class Attention : public mindspore::ops::BaseOperator
MultiHead-Attention op in MindIR.
Public Functions
-
inline Attention()
Constructor.
-
inline void Init() const
Initialize Attention op.
-
inline Attention()