pytorch
95c166cd - Add `is_causal` API for `TransformerDecoder` (#97166)

Commit
1 year ago
Add `is_causal` API for `TransformerDecoder` (#97166) The same API is implemented for `TransformerEncoder`, where this argument is passed through to the sublayers. Pull Request resolved: https://github.com/pytorch/pytorch/pull/97166 Approved by: https://github.com/mikekgfb
Author
Committer
Parents
Loading