Browse Source

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

GitOrigin-RevId: 38349c8357
tags/v1.3.0
Megvii Engine Team 4 years ago
parent
commit
886e7c6eb6
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      imperative/python/src/tensor.cpp

+ 2
- 1
imperative/python/src/tensor.cpp View File

@@ -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(){


Loading…
Cancel
Save