1. Evaluate Expression
      调试过程中,需要查看比如attn_weight 张量tensor的值。
    1. 方法一:attn_weight.detach().numpy(),view as array
    1. 方法二:attn_weight.cpu().numpy(),view as array

在这里插入图片描述
在这里插入图片描述

更多推荐