标签： python tensorflow keras deep-learning
我正在尝试使用TF keras构建NER，当我执行零填充以使批次具有相同的长度，然后添加tf.keras.layers.Masking（）层来屏蔽填充时，我得到的错误是CuDNNLSTM不支持屏蔽。
我的Tensorflow-gpu版本是1.11，Cuda 9.0，CudNN 7.3.1。
我目前确实有一个版本，可以将相同长度的样本分组到每批中以避免填充，我想知道我们是否应该从现在开始不填充，或者CuDNN LSTM将来是否支持屏蔽填充？
谢谢
答案 0 :(得分：1)
刚发现这个
CuDNN RNN暂时无法与掩膜一起使用，并且可能永远不会使用。它需要在CuDNN级别上实现。
请参阅here