随笔分类 -  深度学习 / 动手学深度学习 / 注意力机制 / 多头注意力

10.5.2 实现
摘要：valide_lens是二维数组，假设其如下 tensor([[ 0, 1, 2, 3, 4], [ 5, 6, 7, 8, 9], [10, 11, 12, 13, 14], [15, 16, 17, 18, 19]]) 那么对其在第零维进行repeat_interleave后，如下 tensor 阅读全文
posted @ 2025-02-23 17:26 最爱丁珰 阅读(40) 评论(0) 推荐(0)