pytorch张量，添加权重

如何使张量更重要？例如：也许分配更多的权重

这样做的目的是在邻里聚集期间，对直接邻居给予更多的重视，而对更远的邻居给予更少的重视

我尝试了乘法，但认为我做得不对，因为我只是将所有值乘以2，而不是赋予更多的重要性

aggregate_feats = 2*self.aggregate(nb, pre_hidden_embs, pre_neighs)
cur_hidden_embs = sage_layer(self_feats=pre_hidden_embs[nb],
                                    aggregate_feats=aggregate_feats)

---

知道我该怎么做吗？

欢迎来到Stackoverflow！请发一封邮件，以便我们能帮助你。