class_docstr: A simple Device/CPU Array object in runtime.
docstr: Record the stream that is using this tensor.
  
  Note
  ----
  This API is more for testing. Users should call ``record_stream``
  on torch.Tensor or dgl.graph directly.
  
  Parameters
  ----------
  stream : DGLStreamHandle
docstr_tok: ["Record","the","stream","that","is","using","this","tensor",".","Note","--","--","This","API","is","more","for","testing",".","Users","should","call","``","record_stream","``","on","torch.Tensor","or","dgl.graph","directly",".","Parameters","--","--","--","--","--","stream",":","DGLStreamHandle"]
code: def record_stream(self, stream):
      """Record the stream that is using this tensor.
  
      Note
      ----
      This API is more for testing. Users should call ``record_stream``
      on torch.Tensor or dgl.graph directly.
  
      Parameters
      ----------
      stream : DGLStreamHandle
      """
      check_call(_LIB.DGLArrayRecordStream(self.handle, stream))
  
imports: from __future__ import absolute_import
  import ctypes
  import sys
  import numpy
  from .base import _FFI_MODE, _LIB, c_array, c_str, check_call, string_types
  from .runtime_ctypes import dgl_shape_index_t, DGLArray, DGLArrayHandle, DGLContext, DGLDataType, TypeCode
label_desc: Use image node_id 17 for calling the NDArrayBase obj's underlying member method code with example usage: obj.record_stream(stream) without return types