问题描述
我有 PyTorch 1.7.1，正在查看：https://github.com/pytorch/pytorch/blob/master/torch/utils/mkldnn.py 我没有看到对 cpu 上双向 LSTM 的 MKLDNN 支持。此外，当我详细运行我的代码时，我发现调用只针对 MKL，而不是针对 MKLDNN。我检查了我的 pytorch 配置，显示 MKL 和 MKLDNN 都已开启。
我是否遗漏了什么，或者 Pytorch 中的 cpu 上的双向 LSTM 不支持 MKLDNN？
我正在对 Tensorflow/Keras 和 PyTorch 上的 LSTM 进行比较，所以只是想确保我没有遗漏任何东西。
解决方法
暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！
如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@）