tensorplay.nn.attention.and_masks

tensorplay.nn.attention.and_masks(*mask_mods: Callable[[Tensor, Tensor, Tensor, Tensor], Tensor]) → Callable[[Tensor, Tensor, Tensor, Tensor], Tensor][source]

Returns a mask_mod that’s the intersection of provided mask_mods