Method
__init__
(self, mask_flag=True, factor=1, scale=None, attention_dropout=0.1, output_attention=False)
Source from the content-addressed store, hash-verified
| 17 | """ |
| 18 | |
| 19 | def __init__(self, mask_flag=True, factor=1, scale=None, attention_dropout=0.1, output_attention=False): |
| 20 | super(AutoCorrelation, self).__init__() |
| 21 | self.factor = factor |
| 22 | self.scale = scale |
| 23 | self.mask_flag = mask_flag |
| 24 | self.output_attention = output_attention |
| 25 | self.dropout = nn.Dropout(attention_dropout) |
| 26 | |
| 27 | def time_delay_agg_training(self, values, corr): |
| 28 | """ |
Tested by
no test coverage detected