[Fix] assertion for slices with at least one dim the same size as input dim