TensorFlow RNNCell能否拥有比3D更多的尺寸?

时间:2017-01-09 14:06:42

标签: tensorflow

我想将tf.nn.rnn与视频帧的5D张量(样本,时间,高度,宽度,频道)一起使用,而不必在每个cell.__call__上展平和重塑。这可能吗?

1 个答案:

答案 0 :(得分:0)

从TensorFlow 1.2开始,这将得到支持。也可以通过设置time_major=True并在tf.nn.rnn来电之外相应地重新排序dims来实现。