mirror of
https://github.com/ROCm/jax.git
synced 2025-04-24 14:36:07 +00:00

Change implementation to use native JAX everywhere allowing vmaping and gradients wrt scale and translation.
18 lines
233 B
ReStructuredText
18 lines
233 B
ReStructuredText
jax.image package
|
|
=================
|
|
|
|
.. currentmodule:: jax.image
|
|
|
|
.. automodule:: jax.image
|
|
|
|
|
|
Image manipulation functions
|
|
----------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
resize
|
|
scale_and_translate
|
|
|