enc_output,enc_slf_attn = self.slf_attn(user_embedding,item_input,item_input,mask = slf_attn_mask)
实际上只有一个返回值,但是我却写了两个返回值,所以报错。
改正为
enc_output = self.slf_attn(user_embedding,item_input,item_input,mask = slf_attn_mask)
就好了
写代码的时候发现这个错误,记下以防以后忘记。
本文记录了一个关于自注意力机制实现中的常见错误:当自注意力层仅返回一个输出时,代码中错误地指定了两个返回值。通过调整代码确保返回值数量正确,避免运行时错误。
9548

被折叠的 条评论
为什么被折叠?



