Added tests for non-contiguous 2D and 4D tensors