probly.representation.sample.array_functions

Numpy array function implementations for sample arrays.

Classes

ArraySampleCreator

Protocol for creating sample arrays.

ArraySampleInternals

Internal information about a sample array.

Functions

array_apply_along_axis_function

Implementation of np.apply_along_axis for sample arrays.

array_concatenate_function

Implementation of np.concatenate for sample arrays.

array_copy_function

Implementation of np.copy for sample arrays.

array_expand_dims_function

Implementation of np.expand_dims for sample arrays.

array_function_override

Decorator to convert a bound array function to an array function.

array_internals_override

Decorator to convert a function that takes a call with an array-sample .

array_matrix_transpose

Implementation of np.matrix_transpose for sample arrays.

array_reduction_function

Implementation of dimension-reducing array functions with a keepdims kwarg.

array_reshape_function

Implementation of np.reshape for sample arrays.

array_sample_axis_preserving_function

Implementation of sample-axis-preserving array functions.

array_sample_internals

Get the sample dimension of a sample array.

array_squeeze_function

Implementation of np.squeeze for sample arrays.

array_stack_function

Implementation of np.stack for sample arrays.

array_swapaxes_function

Implementation of np.swapaxes for sample arrays.

array_transpose

Implementation of np.transpose for sample arrays.

track_sample_axis_after_reduction

Track the sample axis after a reduction operation.