输入数据:
接收两个张量:
X_Q:形状为 TQ×dmodel 的 Query 序列,TQ 表示查询序列的长度,dmodel 表示特征维度。X_K:形状为 TK×dmodel 的 Key/Value 序列,TK 表示键值序列的长度。线性变换:
给定 Query 序列
XQ∈RTQ×dmodel和 Key/Value 序列
By signing up a CodeFun2000 universal account, you can submit code and join discussions in all online judging services provided by us.