Skip to content

attention_keras.py中,Position_Embedding的pos的计算 #10

@rivershadowy

Description

@rivershadowy

你好,

想请教一下,attention_keras.py中的第21行

position_i = K.cumsum(K.ones_like(x[:,:,0]), 1)-1 #K.arange不支持变长,只好用这种方法生成

这行代码具体在计算什么?

我翻阅了论文原文和ones_like, cumsum的keras backend资料,对这行代码的用途猜测是计算论文中Position_Embedding公式的pos值,但不太理解具体是如何实现计算的,论文似乎也没有详细涉及。

请问能解释一下吗?谢谢!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions