This repository was archived by the owner on Oct 31, 2025. It is now read-only.
This repository was archived by the owner on Oct 31, 2025. It is now read-only.
Inconsistency in function's doc-string #1794
{{ message }}
Description
Inconsistent function description at https://github.com/google/trax/blob/master/trax/layers/attention.py#L330C1-L342C23
The function states that it "Returns attention-computed per-head activations and unchanged mask." but returns only the activations.