Существует ли какая-либо эквивалентная функция (или некоторая быстрая реализация), которая позволяет мне извлекать 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