小编jul*_*iet的帖子

我有一个大小为 的张量[3, 15, 136]，其中：

• 3 is batch size
• 15 - sequence length和
• 136 is tokens

我想使用维度 (136) 中的概率来独热我的张量tokens。为此，我想提取序列长度中每个字母的标记维度，并将其置于1最大可能性，并将所有其他标记标记为0。