ND-tensor support for accuracy layer
ND-tensor support for argmax layer