modelzoo.transformers.utils.attention_utils.StridedSparseAttentionBuilder#
- class modelzoo.transformers.utils.attention_utils.StridedSparseAttentionBuilder[source]#
Bases:
modelzoo.transformers.utils.attention_utils.SparseAttentionBuilder
Methods
build_mask
get_pytorch_mask
set_global_attention
set_local_attention