probly.representation.sample.array_functions.array_expand_dims_function¶

probly.representation.sample.array_functions.array_expand_dims_function(func: Callable, params: BoundArguments, create_sample: ArraySampleCreator, array: NumpyArrayLike, sample_axis: int, weights: np.ndarray | None) → Any[source]¶: Implementation of np.expand_dims for sample arrays.