Add regression test for decoder_only_block/decoder_only API
Exercises the new Nn_blocks.decoder_only helper with a 2-layer stack,
causal mask, and forward pass, validating output shape. This ensures
the new public API added in the previous commit has CI coverage.
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
ee621a
-
Apr 22 21:44 +00:00