description: Computes indexes into a flattened image for building im2row.
View source on GitHub |
Computes indexes into a flattened image for building im2row
.
tfp.experimental.nn.util.im2row_index(
input_shape, block_shape, rank=2, slice_step=(1, 1), dilations=(1, 1),
dtype=tf.int32, transpose=False, validate_args=False, name=None
)