_merge_masks

pytorch · Nov 14, 2022 · fb355b0 · fb355b0
1 parent ee378ee
commit fb355b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch/nn/modules/transformer.py b/torch/nn/modules/transformer.py
@@ -500,7 +500,7 @@ def forward(self, src: Tensor, src_mask: Optional[Tensor] = None,
                                               "input/output projection weights or biases requires_grad")
 
             if not why_not_sparsity_fast_path:
-                merged_mask, mask_type = F.merge_masks(src_mask, src_key_padding_mask, src)
+                merged_mask, mask_type = F._merge_masks(src_mask, src_key_padding_mask, src)
                 return torch._transformer_encoder_layer_fwd(
                     src,
                     self.self_attn.embed_dim,