This repository was archived by the owner on Oct 16, 2023. It is now read-only.
This repository was archived by the owner on Oct 16, 2023. It is now read-only.
[doc] Add shape annotation for complex layer #8
{{ message }}
For complex layers such as self-attention, we can added shape annoation (e.g.
# hidden states (b, s, h)) so that the other maintainer can better understand the model logic.