精度损失很大

[hermosayhl]

你好！博主，我按照你的教程和代码，终于折腾出来了。但是 C++ 加载之后，推理得到的结果跟 pytorch 的输出相差很大，是不是 kernel 函数写错了？

[hermosayhl]

原来是这个代码没有传张量的长度，已解决。谢谢这位博主的帮助！