ValueError: (InvalidArgument) Illegal index value, Input(input) value should be less than Input(depth), but received input (6) not less than depth (4)
[Hint: Expected p_in_data[i] < depth_, but received p_in_data[i]:6 >= depth_:4.] (at /paddle/paddle/phi/kernels/cpu/one_hot_kernel.cc:58)
[operator < one_hot_v2 > error]