Расширение tf.extract_image_patches до N-мерных патчей - PullRequest
0 голосов
/ 14 марта 2020

Существует ли какая-либо эквивалентная функция (или некоторая быстрая реализация), которая позволяет мне извлекать N-мерные патчи так же, как tf.extract_image_patches и аналогично tf.extract_volume_patches?

Например, я бы хотел иметь что-то вроде этого:

input = #say [10, 12, 15, 15, 12, 7, 5] size tensor
patches = tf.extract_patches(input, ksizes=[1, 3, 3, 3, 3, 1, 1], strides=[1, 3, 3, 3, 3, 1, 1], padding='VALID')
# patches is of size [10/1, 12/3, 15/3, 15/3, 12/3, 7/1, 5/1, 1*3*3*3*3*1*1]
# so the pattern is to dump the patches into a single final dimension according to ksizes
...