tensorflow.keras.layers中有两个关于LSTM的高级API:CuDNNLSTM和LSTM,从名字就可以看出CuDNNLSTM是为使用CUDA并行计算设计的,加速效果从几倍到几十倍。
Tensorflow2.0以上版本在导入CuDNNLSTM会报如题所示错误,改为以下方式导入即可:
from tensorflow.compat.v1.keras.layers import CuDNNLSTM as LSTM
tensorflow.keras.layers中有两个关于LSTM的高级API:CuDNNLSTM和LSTM,从名字就可以看出CuDNNLSTM是为使用CUDA并行计算设计的,加速效果从几倍到几十倍。
Tensorflow2.0以上版本在导入CuDNNLSTM会报如题所示错误,改为以下方式导入即可:
from tensorflow.compat.v1.keras.layers import CuDNNLSTM as LSTM