bert 如何从注意力层获取注意力矩阵输出?

r6hnlfcb  于 4个月前  发布在  其他
关注(0)|答案(1)|浏览(51)

我想了解注意力矩阵,所以想输出它。如何从原始代码中获得输出?我尝试了tf.Session(),但不起作用。我能得到其他方法吗?

qzlgjiam

qzlgjiam1#

我的电子邮件地址是huggingface/transformers#1432。如果你愿意,我们可以成为微信朋友。

相关问题