提交 886e7c6e 编写于 作者: M Megvii Engine Team

refactor(mge/tensor): support scalar in detach and grad

GitOrigin-RevId: 38349c8357871d853219026d1431b98ca55fe3a4
上级 a1ca50c9
......@@ -495,9 +495,10 @@ PyObject* TensorWrapper::detach() {
new_tensor = std::make_shared<Tensor>(m_tensor->m_var);
}
new_tensor->m_trace_info = m_tensor->m_trace_info;
new_tensor->m_flags = m_tensor->m_flags;
auto ret = TensorWrapper::make(pytype, std::move(new_tensor));
return ret.release().ptr();
}
PyObject* TensorWrapper::_dev_tensor(){
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册