numpy_expand_dims_function

probly.representation.sample.numpy_functions.numpy_expand_dims_function(func: Callable, params: BoundArguments, create_sample: NumpySampleCreator, array: NumpyArrayLike, sample_axis: int, weights: np.ndarray | None) → Any

Implementation of np.expand_dims for sample arrays.