spekk.transformations.common

Some common utility functions used by spekk.transformations.

Functions

compose(x, *wrapping_functions)	Apply each f in fs to x.
get_args_for_index(args, in_axes, i)
get_fn_name(f)
getitem_along_axis(x, axis, i)
identity(x)	Return the input unchanged.
map_1_flattened(map_f, flattened_args, ...)