Tensorflow大张量分流到小张量

Question

我有像下面

x = tf.Variable(tf.truncated_normal([batch, input]), stddev=0.1)) 

假设批次= 99，输入= 5的张量，并且我想分成小张量。

如果x低于：

[[1.0, 2.0, 3.0, 4.0, 5.0] 
[2.0, 3.0, 4.0, 5.0, 6.0] 
[3.0, 4.0, 5.0, 6.0, 7.0] 
[4.0, 5.0, 6.0, 7.0, 8.0] 
......................... 
......................... 
......................... 
[44.0, 55.0, 66.0, 77.0, 88.0] 
[55.0, 66.0, 77.0, 88.0, 99.0]] 

我想分成两个张量

[[1.0, 2.0, 3.0, 4.0, 5.0] 
[2.0, 3.0, 4.0, 5.0, 6.0] 
[3.0, 4.0, 5.0, 6.0, 7.0]] 

和

[4.0, 5.0, 6.0, 7.0, 8.0] 
......................... 
......................... 
[44.0, 55.0, 66.0, 77.0, 88.0] 
[55.0, 66.0, 77.0, 88.0, 99.0]] 

我不知道如何使用tf.split分裂行。

Answer 1

一种便利的方法是拨打tf.slice两次。