2次元テンソルをテンソルフロー内の複数のテンソルに動的に分割する

Question

2次元テンソル（行列）を与えられたら、同じサイズのいくつかの小さなテンソルに分割したいと思います。最大プールの前処理と見なすことができます。例えば、2×4の動的desired_size考える

1 2 3 4 5 6 7 8 
2 3 4 5 6 7 8 9 
3 4 5 6 7 8 9 10 
4 5 6 7 8 9 10 11 

は、出力すべきである：

1 2 3 4 
2 3 4 5 

5 6 7 8 
6 7 8 9 

3 4 5 6 
4 5 6 7 

7 8 9 10 
8 9 10 11 

Iしばらくsliceとgatherを研究しました。しかし、私はまだそれを行う方法がわからない。それをどうやって得るか教えていただけますか？前もって感謝します！

Answer 1

あなたはそれがやや冗長に判明していても、tf.extract_image_patchesを使用することができます。

import numpy as np 
import tensorflow as tf 

x = tf.constant(np.arange(8) + np.arange(1,5)[:,np.newaxis]) 
e = tf.extract_image_patches(x[tf.newaxis,:,:,tf.newaxis], 
    [1, 2, 4, 1], [1, 2, 4, 1], [1, 1, 1, 1], padding='VALID') 
e = tf.reshape(e, [-1, 2, 4]) 
sess = tf.InteractiveSession() 
e.eval() 

# returns 
# array([[[ 1, 2, 3, 4], 
#   [ 2, 3, 4, 5]], 
#  [[ 5, 6, 7, 8], 
#   [ 6, 7, 8, 9]], 
#  [[ 3, 4, 5, 6], 
#   [ 4, 5, 6, 7]], 
#  [[ 7, 8, 9, 10], 
#   [ 8, 9, 10, 11]]]) 

Answer 2

は私がtf.split()で結ば：

num_splits = 2 
desired_size = (2, 4) 
A = tf.constant(a) 

C = tf.concat(tf.split(A, desired_size[0], 0),1) 
D = tf.reshape(tf.concat(tf.split(C, num_splits*desired_size[0], 1), 0), (-1, desired_size[0], desired_size[1])) 

#The result 
[[[ 1 2 3 4] 
[ 2 3 4 5]] 

[[ 5 6 7 8] 
[ 6 7 8 9]] 

[[ 3 4 5 6] 
[ 4 5 6 7]] 

[[ 7 8 9 10] 
[ 8 9 10 11]]] 

# For num_splits = 4, desired_size = (2, 2) you get 
[[[ 1 2] 
[ 2 3]] 

[[ 3 4] 
[ 4 5]] 

[[ 5 6] 
[ 6 7]] 

[[ 7 8] 
[ 8 9]] 

[[ 3 4] 
[ 4 5]] 

[[ 5 6] 
[ 6 7]] 

[[ 7 8] 
[ 8 9]] 

[[ 9 10] 
[10 11]]]