code: def _AXWb(A, X, W, b):
      X = mx.nd.dot(X, W.data(X.context))
      Y = mx.nd.dot(A, X.reshape(X.shape[0], -1)).reshape(X.shape)
      return Y + b.data(X.context)
  
imports: import backend
  import dgl
  import dgl.function
  import dgl.nn.mxnet
  import mxnet
  import networkx
  import numpy
  import pytest
  import scipy
  from mxnet import autograd, gluon, nd
  from utils import parametrize_idtype
  from utils.graph_cases import get_cases, random_bipartite, random_dglgraph, random_graph
label_desc: Use image node_id 2 for calling a global function with example usage: _AXWb(A, X, W, b) and returns: unknown