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.
Defined in File attention.h
public mindspore::ops::BaseOperator
MultiHead-Attention op in MindIR.