paddle.nn.attention.flex_attention.and_masks()
返回一个 mask 函数,对输入的多个 mask 函数结果进行逻辑与运算。
mask_mods (Callable) - mask 函数,签名为 mask_mod(b, h, q_idx, kv_idx)。
mask_mod(b, h, q_idx, kv_idx)
Callable:对所有 mask 结果执行逻辑与运算后的 mask 函数。
COPY-FROM: paddle.nn.attention.flex_attention.and_masks