mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixing the naming conventions in signal.py

Browse Source
This commit is contained in:
tilakrayal 2024-06-13 12:21:25 +05:30 committed by GitHub
parent c5c7fa7089
commit 3ef89a2113
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
jax/_src/scipy/signal.py
View File

@ -188,7 +188,7 @@ def convolve(in1: Array, in2: Array, mode: str = 'full', method: str = 'auto',
precision: PrecisionLike = None) -> Array:
"""Convolution of two N-dimensional arrays.
JAX implementation of :func:`jax.scipy.signal.convolve`.
JAX implementation of :func:`scipy.signal.convolve`.
Args:
in1: left-hand input to the convolution.
@ -253,7 +253,7 @@ def convolve2d(in1: Array, in2: Array, mode: str = 'full', boundary: str = 'fill
fillvalue: float = 0, precision: PrecisionLike = None) -> Array:
"""Convolution of two 2-dimensional arrays.
JAX implementation of :func:`jax.scipy.signal.convolve2d`.
JAX implementation of :func:`scipy.signal.convolve2d`.
Args:
in1: left-hand input to the convolution. Must have ``in1.ndim == 2``.
@ -327,7 +327,7 @@ def correlate(in1: Array, in2: Array, mode: str = 'full', method: str = 'auto',
precision: PrecisionLike = None) -> Array:
"""Cross-correlation of two N-dimensional arrays.
JAX implementation of :func:`jax.scipy.signal.correlate`.
JAX implementation of :func:`scipy.signal.correlate`.
Args:
in1: left-hand input to the cross-correlation.
@ -387,7 +387,7 @@ def correlate2d(in1: Array, in2: Array, mode: str = 'full', boundary: str = 'fil
fillvalue: float = 0, precision: PrecisionLike = None) -> Array:
"""Cross-correlation of two 2-dimensional arrays.
JAX implementation of :func:`jax.scipy.signal.correlate2d`.
JAX implementation of :func:`scipy.signal.correlate2d`.
Args:
in1: left-hand input to the cross-correlation. Must have ``in1.ndim == 2``.