array_split.ShapeSplitter.set_split_extents_by_split_size

ShapeSplitter.set_split_extents_by_split_size()[source]

Sets split shape split_shape and split extents (split_begs and split_ends) from values in split_size and split_num_slices_per_axis.