import  mxnet as mx
import mxnet.ndarray as nd
 
img = nd.array([[[[1],[2],[3],[4]],
                [[5],[6],[7],[8]],
                [[9],[10],[11],[12]],
                [[13],[14],[15],[16]]]])

img = nd.concat(img, img, dim=-1)

img = nd.transpose(img, axes=(0, 3, 1, 2))
 
w = nd.ones([5, 2, 3, 3])

b = nd.array([0 for _ in range(5)])