class_docstr: Do affine Transform for modulating attention score.
code: def __call__(self, x):
      scale = self.param(
          "scale", nn.initializers.ones, (1,), jnp.float32
      )
      bias = self.param(
          "bias", nn.initializers.zeros, (1,), jnp.float32
      )
      return x * nn.sigmoid(scale) * 5 + bias
  
imports: import functools
  from typing import Optional, Sequence
  from flax import linen
  import jax
  import jax.numpy
  from scenic.projects.knowledge_visual_language.models import constants
  from t5x.examples.t5 import layers
  from t5x.examples.t5 import network
label_desc: Use image node_id 1 for calling the AffineTransform obj's underlying member method code with example usage: obj.__call__(x) and returns: unknown