未验证 提交 f66a705f 编写于 作者: R RedContritio 提交者: GitHub

update transpose in op_compat.yaml (#55458)

上级 8f8fa38a
......@@ -904,7 +904,7 @@
- op : transpose
args : (Tensor x, int[] perm)
output : Tensor
output : Tensor(out)
infer_meta :
func : TransposeInferMeta
kernel :
......
......@@ -2791,6 +2791,10 @@
- op : transpose (transpose2)
backward : transpose_grad (transpose2_grad)
inputs :
x : X
outputs :
out : Out
attrs:
perm : axis
extra :
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册