我正在尝试使用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/