Bucketize: check input and output dim order match

Author: Gallinator

kernels/portable/cpu/op_bucketize.cpp

@@ -176,6 +176,8 @@ Tensor& bucketize_tensor_out(
       out);
   ET_KERNEL_CHECK(
       context, tensors_have_same_shape(self, out), InvalidArgument, out);
+  ET_KERNEL_CHECK(
+      context, tensors_have_same_dim_order(self, out), InvalidArgument, out);
   ET_KERNEL_CHECK(
       context, tensor_is_realhbf16_type(self), InvalidArgument, out);