未验证 提交 5a69dbcb 编写于 作者: K kangguangli 提交者: GitHub

fix put_along_axis (#55513)

上级 7fc0fed8
...@@ -1700,9 +1700,9 @@ ...@@ -1700,9 +1700,9 @@
optional : boxes_num optional : boxes_num
- backward_op : put_along_axis_grad - backward_op : put_along_axis_grad
forward : put_along_axis (Tensor arr, Tensor indices, Tensor value, int axis, str reduce = "assign") -> Tensor(out) forward : put_along_axis (Tensor arr, Tensor indices, Tensor values, int axis, str reduce = "assign") -> Tensor(out)
args : (Tensor arr, Tensor indices, Tensor out_grad, int axis, str reduce) args : (Tensor arr, Tensor indices, Tensor out_grad, int axis, str reduce)
output : Tensor(arr_grad), Tensor(value_grad) output : Tensor(arr_grad), Tensor(values_grad)
infer_meta : infer_meta :
func : GeneralBinaryGradInferMeta func : GeneralBinaryGradInferMeta
param : [arr, indices] param : [arr, indices]
......
...@@ -137,6 +137,7 @@ test_pool_max_op ...@@ -137,6 +137,7 @@ test_pool_max_op
test_prelu_op test_prelu_op
test_prior_box_op test_prior_box_op
test_psroi_pool_op test_psroi_pool_op
test_put_along_axis_op
test_range test_range
test_reverse_op test_reverse_op
test_roi_align_op test_roi_align_op
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册