如何在model()中使用maskingLayer

时间:2019-05-08 14:15:05

标签: recurrent-neural-network masking tensorflow.js

我有这个张量流图,就像A,B->串联-> RNN-> timeDistribute->密集-> softmax(最后有点多余)

我还要介绍两件事:1)并非所有时间步都填充有有效数据,因此我想添加一个tf.layers.masking层...我是否必须将其合并到串联中?它有自己的true / false位流(尽管可以从输入A导出)。

另一件事是在输出端屏蔽;因为只有某些动作是可能的,所以我想禁用这些动作的更新/反向考虑...但是我不知道如何将侧面加载的信息添加到最后一个softmax中...

A和B实际上是连接前的一小段操作。

https://gist.github.com/d3x0r/26b713af9d3bcff7e34a21e4c8d04907

我是否必须将遮罩层合并到concat中,而只是“忽略”最后一点?还是可以仅引用另一个“输入C-> masking()”,而不将输出附加到图形上?

0 个答案:

没有答案