tensorplay.nn.attention.noop_mask

tensorplay.nn.attention.noop_mask(batch: Tensor, head: Tensor, token_q: Tensor, token_kv: Tensor) → Tensor[source]

Returns a noop mask_mod