Get Started
Advanced Materials
API Reference
in_feats (int) – Input feature size.
out_feats (int) – Output feature size.
num_heads (int) – Number of heads in Multi-Head Attention.
dropout (float, optional) – Dropout rate, defaults: 0.
0