cellflow.networks.SelfAttentionBlock.takes_attention_mask

SelfAttentionBlock.takes_attention_mask: ClassVar[bool] = True

This block consumes an attention mask (see __call__()), so _apply_modules() forwards attention_mask to it.