如何解决PyTorch 中对双向 LSTM 的 MKLDNN 支持
我有 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 举报,一经查实,本站将立刻删除。