class_docstr: Adaptive Computation Time Function to help we use nn.scan on ACT.
class_docstr_tok: ["Adaptive","Computation","Time","Function","to","help","we","use","nn.scan","on","ACT","."]
code: def skip_a_step(self, x) -> Any:  # Shunt
      return x
  
file_path: scenic/scenic/projects/baselines/universal_transformer/layers.py
imports: from typing import Any
  from flax import linen
  import jax
  import jax.numpy
  import ml_collections
label_desc: Use image node_id 3 for calling the ACTFunction obj's underlying member method code with example usage: obj.skip_a_step(x) and returns: x