Class Attention

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.