code: def normal_init(
      ops: Ops, shape: Shape, *, mean: float = 0
  ) -> FloatsXd:
      size = int(ops.xp.prod(ops.xp.asarray(shape)))
      inits = cast(
          FloatsXd,
          numpy.random.normal(scale=mean, size=size).astype("float32"),
      )
      inits = ops.reshape_f(inits, shape)
      return ops.asarray_f(inits)
  
imports: from typing import Callable, cast
  import numpy
  from .backends import Ops
  from .config import registry
  from .types import FloatsXd, Shape
  from .util import partial
label_desc: Use image node_id 17 for calling a global function with example usage: normal_init(ops, shape) and returns: ops