Update attn.py · e823b31040 - Prediction-MindSpore - 开源协同云脑生态支撑系统

Update attn.py

main

BBing 2 years ago

parent

d7a9768fe5

commit

e823b31040

1 changed files with 1 additions and 1 deletions

Unified View

							
								+ 1
								
								- 1
							
models/attn.pyView File
								
																@@ -4,7 +4,7 @@ import mindspore.ops as ops
															
																from mindspore import Tensor
																
																from mindspore import Tensor
															
																from mindspore.common import dtype as mstype
																
																from mindspore.common import dtype as mstype
															
																1
															
																class FullAttention(nn.Module):
																
																class FullAttention(nn.Module):
															
																    def __init__(self, mask_flag=True, factor=5, scale=None, attention_dropout=0.1, output_attention=False):
																
																    def __init__(self, mask_flag=True, factor=5, scale=None, attention_dropout=0.1, output_attention=False):
															
																        super(FullAttention, self).__init__()
																
																        super(FullAttention, self).__init__()