Triton是一种基于Python的并行编程语言和编译器,用于高效编写自定义DNN计算内核,并能在现代GPU上以最大吞吐量运行。其triton.language.exp(x)函数可计算输入块x的逐元素指数值。
作者尝试使用加权随机函数从一组数值中按权重随机选择较小的子集,但效果不佳。作者尝试使用指数函数放大较大的值,但权重效果不明显。作者怀疑小值太多可能掩盖了权重效果,于是尝试使用对数函数减小差异,然后再次使用指数函数放大较大的值,但选择仍未正确按权重分布。作者想知道问题出在哪里。
完成下面两步后,将自动完成登录并继续当前操作。