目录
一、前置演示:scatter_ 如何“一键”生成 one-hot 与掩码
二、DKD 核心思想(Decoupled Knowledge Distillation)
一、前置演示:scatter_ 如何“一键”生成 one-hot 与掩码
Python
复制
label = torch.LongTensor(batch_size, 1).random_() % class_num
xx = torch.zeros(batch_size, class_num).scatter_(1, label, 1)
该博客介绍了在深度学习中使用蒸馏算法的过程,通过示例展示了如何生成目标掩码,并提供了相关代码实现。主要涉及到张量操作,如scatter_函数的应用。
订阅专栏 解锁全文
267

被折叠的 条评论
为什么被折叠?



