zero_initializer# penzai.nn.linear_and_affine.zero_initializer(key: jax.Array, *, input_axes: dict[str, int], output_axes: dict[str, int], parallel_axes: dict[str, int], convolution_spatial_axes: dict[str, int], dtype: jax.typing.DTypeLike) → jax.Array[source]# Zeros initializer for named arrays.