code: def __init__(self, gidxA, gidxB, num_vtypes):
      super().__init__()
      self.gidxA = gidxA
      self.gidxB = gidxB
      self.num_vtypes = num_vtypes
  
imports: import mxnet
  import numpy
  from mxnet import nd
  from ..._sparse_ops import _bwd_segment_cmp, _csrmask, _csrmm, _csrsum, _gsddmm, _gspmm, _scatter_add, _segment_reduce
  from ...base import ALL, dgl_warning, is_all
  from ...heterograph_index import create_unitgraph_from_csr
  from .tensor import asnumpy, context, copy_to, to_backend_ctx, zerocopy_from_numpy
label_desc: Use image node_id 1 to create a new CSRMM object from inherited base classes: mx.autograd with example: obj = CSRMM(gidxA, gidxB, num_vtypes)