transpose

dragon.vm.tensorflow.transpose(
  a,
  perm=None,
  name=None
)[source]

Permute the dimensions of input.

Examples:

# Provide the permutation for all axes
x = tf.ones(shape=(2, 3, 4))
print(tf.transpose(x, (0, 2, 1)).shape)  # (2, 4, 3)

# Or dimensions will be simply inverse
print(tf.transpose(x).shape)  # (4, 3, 2)
Parameters:
  • a (dragon.Tensor) – The input tensor.
  • perm (Sequence[Union[int, dragon.Tensor]]) – The output permutation.
  • name (str, optional) – A optional name for the operation.
Returns:

dragon.Tensor – The output tensor.