The Infinite Limit is ill-defined when the Input Kernel has `channel_axis == 1`

[DarrenZhang01]

line 190, in new_kernel_fn
    raise ValueError(f'{kernel_fn} requires `{key} == {v_pos}`, '
ValueError: <function Dense.<locals>.kernel_fn at 0x15815e790> requires `channel_axis == 0`, but input kernel has `channel_axis == 1`, making the infinite limit ill-defined.

[DarrenZhang01]

Solved in 98be634, together with issues google/neural-tangents#52, #11 and #6