space_to_depth

tfsnippet.ops.space_to_depth(input, block_size, channels_last=True, name=None)

Wraps tf.space_to_depth(), to support tensors higher than 4-d.

Parameters:

• input – The input tensor, at least 4-d.
• block_size (int) – An int >= 2, the size of the spatial block.
• channels_last (bool) – Whether or not the channels axis is the last axis in the input tensor?
• name (str) – Default name of the name scope. If not specified, generate one according to the method name.

Returns:

The output tensor.

Return type:

tf.Tensor

See also

tf.space_to_depth()