Skip to content

Commit

Permalink
Update docstring description of axis keyword in comulative functions
Browse files Browse the repository at this point in the history
  • Loading branch information
antonwolfy committed May 10, 2024
1 parent 1243edc commit 05426b5
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions dpctl/tensor/_accumulation.py
Original file line number Diff line number Diff line change
Expand Up @@ -229,7 +229,7 @@ def cumulative_sum(
Args:
x (usm_ndarray):
input array.
axis (Optional[int, Tuple[int, ...]]):
axis (Optional[int]):
axis along which cumulative sum must be computed.
If `None`, the sum is computed over the entire array.
If `x` is a one-dimensional array, providing an `axis` is optional;
Expand Down Expand Up @@ -308,7 +308,7 @@ def cumulative_prod(
Args:
x (usm_ndarray):
input array.
axis (Optional[int, Tuple[int, ...]]):
axis (Optional[int]):
axis along which cumulative product must be computed.
If `None`, the product is computed over the entire array.
If `x` is a one-dimensional array, providing an `axis` is optional;
Expand Down Expand Up @@ -388,7 +388,7 @@ def cumulative_logsumexp(
Args:
x (usm_ndarray):
input array.
axis (Optional[int, Tuple[int, ...]]):
axis (Optional[int]):
axis along which cumulative logsumexp must be computed.
If `None`, the logsumexp is computed over the entire array.
If `x` is a one-dimensional array, providing an `axis` is optional;
Expand Down

0 comments on commit 05426b5

Please sign in to comment.