load model now handles ND-tensor in a backward compatible way
update imagenet example for ND-tensor