Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Adding multiple Statistical functions to paddle backend #10869

Merged
merged 11 commits into from
Feb 25, 2023
Merged

Adding multiple Statistical functions to paddle backend #10869

Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
040b8d2
Added prod from Statistical to paddle backend
KevinUnadkat Feb 23, 2023
a85ab52
adding sum , var , std from statistical to paddle backend
KevinUnadkat Feb 23, 2023
5f7a951
updated var , sum , std and also added cumsum , cumprod , einsum
KevinUnadkat Feb 23, 2023
b3ce044
updated cumprod ,var , cumsum , std
KevinUnadkat Feb 24, 2023
adb3af8
updated std , var , cumpord
KevinUnadkat Feb 25, 2023
a1820ee
removed cumsum prod einsum from this pr
KevinUnadkat Feb 25, 2023
d1e8eb3
Update statistical.py
MahmoudAshraf97 Feb 25, 2023
7b83522
made the chnages to var
KevinUnadkat Feb 25, 2023
1d4d82c
made some changes to var
KevinUnadkat Feb 25, 2023
aa6ff90
Update statistical.py
MahmoudAshraf97 Feb 25, 2023
5e5d025
Update statistical.py
MahmoudAshraf97 Feb 25, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
51 changes: 46 additions & 5 deletions ivy/functional/backends/paddle/statistical.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,7 @@ def prod(
keepdims: bool = False,
out: Optional[paddle.Tensor] = None,
) -> paddle.Tensor:
raise IvyNotImplementedException()
return paddle.prod(x, axis=axis, keepdim=keepdims, dtype=dtype)


def std(
Expand All @@ -72,8 +72,27 @@ def std(
keepdims: bool = False,
out: Optional[paddle.Tensor] = None,
) -> paddle.Tensor:
raise IvyNotImplementedException()

if axis is None:
axis = list(range(len(x.shape)))
if axis == ():
return x
axis = (axis,) if isinstance(axis, int) else tuple(axis)
if correction == 0:
return paddle.std(x, axis=axis, unbiased=False, keepdim=keepdims)
elif correction == 1:
return paddle.std(x, axis=axis, unbiased=True, keepdim=keepdims)
size = 1
for a in axis:
size *= x.shape[a]
if size - correction <= 0:
ret = paddle.std(x, axis=axis, unbiased=False, keepdim=keepdims)
ret = ivy.full(ret.shape, float("nan"), dtype=ret.dtype)
return ret
ret = paddle.mul(
paddle.std(x, axis=axis, unbiased=False, keepdim=keepdims),
(size / (size - correction)) ** 0.5,
)
return ret

def sum(
x: paddle.Tensor,
Expand All @@ -84,7 +103,7 @@ def sum(
keepdims: bool = False,
out: Optional[paddle.Tensor] = None,
) -> paddle.Tensor:
raise IvyNotImplementedException()
return paddle.sum(x, axis=axis, dtype=dtype, keepdim=keepdims)


def var(
Expand All @@ -96,7 +115,29 @@ def var(
keepdims: bool = False,
out: Optional[paddle.Tensor] = None,
) -> paddle.Tensor:
raise IvyNotImplementedException()
if axis is None:
axis = list(range(len(x.shape)))
if axis == ():
return x
axis = (axis,) if isinstance(axis, int) else tuple(axis)
if correction == 0:
return paddle.var(x, axis=axis, unbiased=False, keepdim=keepdims)
elif correction == 1:
return paddle.var(x, axis=axis, unbiased=True, keepdim=keepdims)
size = 1
for a in axis:
size *= x.shape[a]
if size - correction <= 0:
ret = paddle.var(x, axis=axis, unbiased=False, keepdim=keepdims)
ret = ivy.full(ret.shape, float("nan"), dtype=ret.dtype)
return ret
else:
ret = paddle.mul(
paddle.var(x, axis=axis, unbiased=False, keepdim=keepdims),
(size / (size - correction)) ** 0.5,
)
return ret



# Extra #
Expand Down