self-attention

self-attention

文章 0 浏览 0

在 PyTorch 中实现 1D 自注意力

我正在尝试使用 PyTorch 实现下面的一维自注意力块: 在以下论文。您可以在下面找到我的(临时)尝试: import torch.nn as nn import torch #INPUT s…

A君 2025-01-15 19:33:58 1 0

mini_batch中的不同节点数

我对图神经网络相当陌生,我正在使用自注意力训练 GNN 模型,我有几个问题。 问题是我的节点数和 node_num 在每个批次中都不同,因此在第一批中我有:…

倾城°AllureLove 2025-01-11 10:16:39 2 0
更多

推荐作者

饮湿

文章 0 评论 0

明月

文章 0 评论 0

02

文章 0 评论 0

hs1283

文章 0 评论 0

风向决定发型

文章 0 评论 0

落花浅忆

文章 0 评论 0

    我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
    原文