Tensorflow RNN：两种不同类型的输入

Question

我想为我的LSTM RNN Cell提供2种类型的输入。我的输入由整数列表组成（即[5,2,3,4,6,1,0, ...]）。

但是，每个整数分为2个不同的组，因此我想以[[5,True],[2,False], [3,False], [4,True], ... ]的方式标记每个整数。

我见过输入由相同类型的常量组成的情况，输入维度为2或更高。但我不确定2种不同类型是否可以产生1个输入单位，如[5,True]。如果这是不可能的，我正在考虑将True替换为整数2，将False替换为整数1，所以像[[5,2], [2,1], ...]，其中输入维度是2（不确定这是标记的好方法）。

什么是标记每个整数以便每个整数属于不同组的好方法？

Answer 1

TensorFlow支持嵌套元组作为rnn输入，请参阅doc。但是，您需要编写自己的单元类来处理此特定类型的输入。在这种情况下，它应该是这样的：

TypeError: can't pickle memoryview objects