我正在尝试使用TF keras构建NER,当我执行零填充以使批次具有相同的长度,然后添加tf.keras.layers.Masking()层来屏蔽填充时,我得到的错误是CuDNNLSTM不支持遮罩。

我的Tensorflow-gpu版本是1.11,Cuda 9.0,CudNN 7.3.1。

我目前确实有版本,可以将相同长度的样本分组到每批中以避免填充,我想知道我们是否应该从现在开始不填充,或者CuDNN LSTM将来是否支持掩蔽填充?

谢谢

最佳答案

刚发现这个


  CuDNN RNN暂时无法与掩膜一起使用,并且可能永远不会使用。它需要在CuDNN级别上实施。


here

关于python - Tensorflow Keras CuDNN LSTM层不支持 mask ,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/53194564/

10-12 22:11