Method forward

LDPS_Graph/layers/PatchTST_backbone.py:193–205

(self, src:Tensor, 
                edge_index: torch.LongTensor,
                edge_weight: torch.FloatTensor,
                key_padding_mask:Optional[Tensor]=None, 
                attn_mask:Optional[Tensor]=None)

        self.res_attention = res_attention

    def forward(self, src:Tensor, 
                edge_index: torch.LongTensor,
                edge_weight: torch.FloatTensor,
                key_padding_mask:Optional[Tensor]=None, 
                attn_mask:Optional[Tensor]=None):
        output = src
        scores = None
        if self.res_attention:
            for mod in self.layers: output, scores = mod(output,edge_index, edge_weight,  prev=scores, key_padding_mask=key_padding_mask, attn_mask=attn_mask)
            return output
        else:
            for mod in self.layers: output = mod(output, key_padding_mask=key_padding_mask, attn_mask=attn_mask)
            return output


### Add DG-Conv and SG-Conv