| 732 | } |
| 733 | |
| 734 | PyObject* TensorWrapper::_dev_tensor() { |
| 735 | auto dv = m_tensor->data().dev_tensor(); |
| 736 | // TODO: handle scalar |
| 737 | return py::cast(dv->as_nd(true)).release().ptr(); |
| 738 | } |
| 739 | |
| 740 | void TensorWrapper::_drop() { |
| 741 | imperative::apply(DTRCommand(DTRCommand::Drop), m_tensor->data()); |