class_docstr: Shortcut in ResNet.
  
  Attributes:
    features: Number of features.
    stride: Stride of the down-sampled output.
class_docstr_tok: ["Shortcut","in","ResNet",".","Attributes",":","features",":","Number","of","features",".","stride",":","Stride","of","the","down-sampled","output","."]
code: def __call__(self, x: jnp.ndarray) -> jnp.ndarray:
      x = nn.avg_pool(
          x, (self.stride, self.stride), (self.stride, self.stride)
      )
      x = nn.Conv(
          self.features,
          (1, 1),
          strides=(1, 1),
          use_bias=False,
          name="0",
      )(x)
      x = nn.BatchNorm(use_running_average=True, name="1")(x)
      return x
  
imports: import functools
  from typing import Sequence, Optional, Union, Tuple
  import flax.linen
  import jax
  import jax.numpy
label_desc: Use image node_id 1 for calling the Shortcut obj's underlying member method code with example usage: obj.__call__(x) and returns: x