check_unique_axis_names_for_initializer

check_unique_axis_names_for_initializer#

penzai.nn.linear_and_affine.check_unique_axis_names_for_initializer(input_axes: dict[str, int], output_axes: dict[str, int], parallel_axes: dict[str, int], convolution_spatial_axes: dict[str, int]) None[source]#

Checks that indices appear exactly once in the given axis specifications.