gradient_image

ivy.gradient_image(x, f=None)[source]

Computes image gradients (dy, dx) for each channel.

Parameters
  • x (array) – Input image [batch_shape, h, w, d] .

  • f (ml_framework, optional) – Machine learning library. Inferred from inputs if None.

Returns

Gradient images dy [batch_shape,h,w,d] and dx [batch_shape,h,w,d] .


Supported Frameworks:

empty jax_logo empty tf_logo empty pytorch_logo empty mxnet_logo empty numpy_logo empty