Skip to content

Commit 60a8ad7

Browse files
committed
Corrected prod docstring
Small tweaks to sum, min, and max docstrings
1 parent 8890d21 commit 60a8ad7

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

dpctl/tensor/_reduction.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ def sum(x, axis=None, dtype=None, keepdims=False):
149149
Args:
150150
x (usm_ndarray):
151151
input array.
152-
axis (Optional[int, Tuple[int,...]]):
152+
axis (Optional[int, Tuple[int, ...]]):
153153
axis or axes along which sums must be computed. If a tuple
154154
of unique integers, sums are computed over multiple axes.
155155
If `None`, the sum is computed over the entire array.
@@ -210,10 +210,10 @@ def prod(x, axis=None, dtype=None, keepdims=False):
210210
Args:
211211
x (usm_ndarray):
212212
input array.
213-
axis (Optional[int, Tuple[int,...]]):
214-
axis or axes along which sums must be computed. If a tuple
215-
of unique integers, sums are computed over multiple axes.
216-
If `None`, the sum is computed over the entire array.
213+
axis (Optional[int, Tuple[int, ...]]):
214+
axis or axes along which products must be computed. If a tuple
215+
of unique integers, products are computed over multiple axes.
216+
If `None`, the product is computed over the entire array.
217217
Default: `None`.
218218
dtype (Optional[dtype]):
219219
data type of the returned array. If `None`, the default data
@@ -237,7 +237,7 @@ def prod(x, axis=None, dtype=None, keepdims=False):
237237
where input array `x` is allocated.
238238
If the data type (either specified or resolved) differs from the
239239
data type of `x`, the input array elements are cast to the
240-
specified data type before computing the sum. Default: `None`.
240+
specified data type before computing the product. Default: `None`.
241241
keepdims (Optional[bool]):
242242
if `True`, the reduced axes (dimensions) are included in the result
243243
as singleton dimensions, so that the returned array remains
@@ -314,7 +314,7 @@ def max(x, axis=None, keepdims=False):
314314
Args:
315315
x (usm_ndarray):
316316
input array.
317-
axis (Optional[int, Tuple[int,...]]):
317+
axis (Optional[int, Tuple[int, ...]]):
318318
axis or axes along which maxima must be computed. If a tuple
319319
of unique integers, the maxima are computed over multiple axes.
320320
If `None`, the max is computed over the entire array.
@@ -342,7 +342,7 @@ def min(x, axis=None, keepdims=False):
342342
Args:
343343
x (usm_ndarray):
344344
input array.
345-
axis (Optional[int, Tuple[int,...]]):
345+
axis (Optional[int, Tuple[int, ...]]):
346346
axis or axes along which minima must be computed. If a tuple
347347
of unique integers, the minima are computed over multiple axes.
348348
If `None`, the min is computed over the entire array.

0 commit comments

Comments
 (0)