问题描述
我正在玩GPT2，我有2个张量：
O ：形状为（B，S-1，V）的输出张量，其中B是批处理大小，S是时间步数，V是词汇量。这是生成模型的输出，并且在第二维上被软最大化。
L ：2D张量形状（B，S-1），其中每个元素是每个样本的每个时间步的正确标记的索引。这基本上是标签。
我想基于张量 L 从张量 O 中提取相应正确令牌的预测概率，以使最终得到2D张量形状（B， S）。除了使用循环之外，还有一种有效的方法吗？
解决方法
暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！
如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@）