code: def _S2AXWb(A, N, X, W, b):
      X1 = X * N
      X1 = mx.nd.dot(A, X1.reshape(X1.shape[0], -1))
      X1 = X1 * N
      X2 = X1 * N
      X2 = mx.nd.dot(A, X2.reshape(X2.shape[0], -1))
      X2 = X2 * N
      X = mx.nd.concat(X, X1, X2, dim=-1)
      Y = mx.nd.dot(X, W)
  
      return Y + b
  
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 6 for calling a global function with example usage: _S2AXWb(A, N, X, W, b) and returns: unknown