#186 add tensor.chunk(), add some math functions

Merged
zoulq merged 3 commits from tanh01 into master 1 year ago
kcl commented 1 year ago
zoulq reviewed 1 year ago
@@ -1842,0 +1843,4 @@

def hypot(input, other, *, out=None):
#TODO: can't found ms.ops.hypot()
if out is not None:
zoulq commented 1 year ago
out统一整改已经上库,可以调用_out_inplace_assign()
zoulq reviewed 1 year ago
@@ -1842,0 +1853,4 @@


def i0(input, *, out=None):
if out is not None:
zoulq commented 1 year ago
同上
zoulq reviewed 1 year ago
@@ -1842,0 +1909,4 @@
if input.dtype == ms.float16:
input = input.astype(ms.float32)
other = other.astype(ms.float32)
output = ms.ops.igammac(input, other).astype(ms.float16)
zoulq commented 1 year ago
这一段代码逻辑和上一个函数一样,可以抽取出一个公共的内部函数
zoulq merged commit db7bc5b72a into master 1 year ago
The pull request has been merged as db7bc5b72a.
Sign in to join this conversation.
No reviewers
No Label
No Milestone
No Assignees
2 Participants
Notifications
Due Date

No due date set.

Dependencies

This pull request currently doesn't have any dependencies.

Loading…
There is no content yet.