0

我想使用 tensorflow 将序列数据训练为带有一些零填充的 Rnn 基本模型。

我希望模型在训练时忽略 0 值。

我是否必须设置参数才能做到这一点？还是模型会自动忽略零？

谢谢，，

4

1 回答 1

0

它不是自动的，你应该引入Masking来实现这一点。这意味着层如何能够知道何时忽略序列输入中的某些时间步。

你可以通过三种方式介绍它

  • 您可以添加一个tf.keras.layers.Masking图层
  • 您可以配置一个tf.keras.layers.Embedding层mask_zero=True
  • 您可以在调用支持该参数的层时手动传递mask参数

有关更多信息，您可以参考使用 Keras 的 Masking and padding指南。

于 2021-06-01T16:41:20.010 回答