transpose_conv2d_axis

tfsnippet.ops.transpose_conv2d_axis(input, from_channels_last, to_channels_last, name=None)

Ensure the channels axis of input tensor to be placed at the desired axis.

Parameters:

• input (tf.Tensor) – The input tensor, at least 4-d.
• from_channels_last (bool) – Whether or not the channels axis is the last axis in input? (i.e., the data format is “NHWC”)
• to_channels_last (bool) – Whether or not the channels axis should be the last axis in the output tensor?
• name (str) – Default name of the name scope. If not specified, generate one according to the method name.

Returns:

The (maybe) transposed output tensor.

Return type:

tf.Tensor