rocm_jax/docs/jax.image.rst

``jax.image`` module
====================

.. currentmodule:: jax.image

.. automodule:: jax.image


Image manipulation functions
----------------------------

.. autosummary::
  :toctree: _autosummary

    resize
    scale_and_translate

Argument classes
----------------

.. currentmodule:: jax.image

.. autoclass:: ResizeMethod
DOC: make API doc titles more uniform 2023-01-18 10:59:42 -08:00			``jax.image`` module
			`====================`
Add jax.image.resize. (#3703) * Add jax.image.resize. This is a port of `tf.image.resize()` and the `ScaleAndTranslate` operator. While I don't expect this implementation to be particularly fast, it is a useful generic implementation to which we can add optimized special cases as the need arises. 2020-07-10 09:57:59 -04:00
			`.. currentmodule:: jax.image`

			`.. automodule:: jax.image`


			`Image manipulation functions`
			`----------------------------`

			`.. autosummary::`
			`:toctree: _autosummary`

			`resize`
Expose scale_and_translate as a public function, fix a bug in implementation when translation is not 0. Change implementation to use native JAX everywhere allowing vmaping and gradients wrt scale and translation. 2020-09-08 13:51:19 -07:00			`scale_and_translate`
Add jax.image.resize. (#3703) * Add jax.image.resize. This is a port of `tf.image.resize()` and the `ScaleAndTranslate` operator. While I don't expect this implementation to be particularly fast, it is a useful generic implementation to which we can add optimized special cases as the need arises. 2020-07-10 09:57:59 -04:00
Documentation improvements. 2022-03-08 09:35:36 -05:00			`Argument classes`
			`----------------`

			`.. currentmodule:: jax.image`

			`.. autoclass:: ResizeMethod`