-
Notifications
You must be signed in to change notification settings - Fork 5.8k
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.
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
Added torch.ge
and torch.greater_equal
to the frontend
#3637
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just me being super picky, but small change and should be good to go :)
@@ -49,3 +49,11 @@ def eq(input, other, *, out=None): | |||
|
|||
def argsort(input, dim=-1, descending=False): | |||
return ivy.argsort(input, axis=dim, descending=descending) | |||
|
|||
|
|||
def ge(input, other, *, out=None): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This is me being really picky but could you set greater_equal to be the main function and ge to be the alias. I think its more readable :)
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Yeah makes sense Let me do it I only implemented this way following torch implementation
@@ -175,3 +175,44 @@ def test_torch_argsort( | |||
dim=axis, | |||
descending=descending, | |||
) | |||
|
|||
|
|||
# ge |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Same here
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Once passing will merge :)
Hi
This PR resolves issues #3620 and #3621
These two functions
torch.ge
andtorch.greater_equal
are aliases.Thanks