Function rot90

tensorflow/python/ops/image_ops_impl.py:488–524

Rotate image(s) counter-clockwise by 90 degrees. For example: ```python a=tf.constant([[[1],[2]],[[3],[4]]]) # rotating `a` counter clockwise by 90 degrees a_rot=tf.image.rot90(a,k=1) #rotated `a` print(a_rot) # [[[2],[4]],[[1],[3]]] ``` Args: image: 4-D Tensor of shape `[batch

(image, k=1, name=None)

@tf_export('image.rot90')
def rot90(image, k=1, name=None):
  """Rotate image(s) counter-clockwise by 90 degrees.


  For example:
  ```python
  a=tf.constant([[[1],[2]],[[3],[4]]])
  # rotating `a` counter clockwise by 90 degrees
  a_rot=tf.image.rot90(a,k=1) #rotated `a`
  print(a_rot) # [[[2],[4]],[[1],[3]]]
  ```
  Args:
    image: 4-D Tensor of shape `[batch, height, width, channels]` or 3-D Tensor
      of shape `[height, width, channels]`.
    k: A scalar integer. The number of times the image is rotated by 90 degrees.
    name: A name for this operation (optional).

  Returns:
    A rotated tensor of the same type and shape as `image`.

  Raises:
    ValueError: if the shape of `image` not supported.
  """
  with ops.name_scope(name, 'rot90', [image, k]) as scope:
    image = ops.convert_to_tensor(image, name='image')
    image = _AssertAtLeast3DImage(image)
    k = ops.convert_to_tensor(k, dtype=dtypes.int32, name='k')
    k.get_shape().assert_has_rank(0)
    k = math_ops.mod(k, 4)

    shape = image.get_shape()
    if shape.ndims == 3 or shape.ndims is None:
      return _rot90_3D(image, k, scope)
    elif shape.ndims == 4:
      return _rot90_4D(image, k, scope)
    else:
      raise ValueError('\'image\' must have either 3 or 4 dimensions.')


def _rot90_3D(image, k, name_scope):