code: def __call__(self, inputs):
      kernel = self.param(
          "kernel", self.kernel_init, (inputs.shape[-1], self.features)
      )
      y = lax.dot_general(
          inputs,
          kernel,
          (((inputs.ndim - 1,), (0,)), ((), ())),
      )
      if self.use_bias:
          bias = self.param("bias", self.bias_init, (self.features,))
          y = y + bias
      return y
  
imports: from jax import lax
  from flax.linen import initializers
  from typing import Callable
  from flax.linen import Module, compact
label_desc: Use image node_id 1 for calling the Dense obj's underlying member method code with example usage: obj.__call__(inputs) and returns: y