# put_along_axis — CPU crash
# 根因: values tensor 含 0-size 维度

# case 1: values axis-0 = 0
put_along_axis(arr=Tensor.uint8((10, 10, 10)), indices=Tensor.int64((5, 5, 5)), values=Tensor.uint8((0, 5, 5)), axis=1, reduce="mul", include_self=True, broadcast=False)

# case 2: values axis-2 = 0
put_along_axis(arr=Tensor.uint8((10, 10, 10)), indices=Tensor.int64((5, 5, 5)), values=Tensor.uint8((5, 5, 0)), axis=1, reduce="mul", include_self=True, broadcast=False)
