pytorch 如何更加重视聚合特征、注意力机制

Question

    def forward(self, nodes_batch):
            """
...
            #Initiate self feature of center node
            pre_hidden_embs = self.raw_features
            for index in range(1, self.num_layers+1):
                #nb = lower_layer_nodes of 1st order , followed by 2nd order
                nb = nodes_batch_layers[index][0]
                #Extract 3 tuples from 2nd order , followed by 1st
                pre_neighs = nodes_batch_layers[index-1]
                # self.dc.logger.info('aggregate_feats.')
                #Aggregate 1st layer unique nodes, self , 2nd layer unique nodes/direct neighs + self
                #Aggregate 2nd layer unique nodes,self,
    
                aggregate_feats = self.aggregate(nb, pre_hidden_embs, pre_neighs)
                sage_layer = getattr(self, 'sage_layer'+str(index))
                if index > 1:
                    #_node_map returns index of lower_layer_nodes_dict --> unique center nodes of layer 0
                    nb = self._nodes_map(nb, pre_hidden_embs, pre_neighs)
                    **#aggregate_feats = 2*self.aggregate(nb, pre_hidden_embs, pre_neighs)**
                # self.dc.logger.info('sage_layer.')
                #W/ nb index , retrieve self + aggregate_feats embeddings (2nd order +1st , then 1st + zero layers)
                cur_hidden_embs = sage_layer(self_feats=pre_hidden_embs[nb],
                                            aggregate_feats=aggregate_feats)
                #Agg neigh of 2nd layer to 1st layer
                #Then aggreg 1st layer(aggregated earlier) to zero layer
                #From outside to inside
                pre_hidden_embs = cur_hidden_embs
    
            return pre_hidden_embs

请注意以下已停用的内容，如果它是第一层到中心节点，我计划为这个aggregate_feats分配更多权重，如果它是第二层到第一层，则为aggregate_feats分配更少权重。我可以知道如何实现这一目标吗？ #aggregate_feats = 2*self.aggregate(nb, pre_hidden_​​embs, pre_neighs)

换句话说，我如何为某个嵌入分配更多的权重，在这种情况下，它是aggregate_feats。