()
return tl.Fn('AsTokenIDs', lambda x: x.astype(jnp.int32))
Returns a layer that makes mask values look like token ID ints.
Returns a layer that makes mask values look like token ID ints.
Returns a layer that makes mask values look like token ID ints .
def _AsTokenIDs():
  """Returns a layer that makes mask values look like token ID ints."""
  return tl.Fn('AsTokenIDs', lambda x: x.astype(jnp.int32))
def _AsTokenIDs ( ) : return tl . Fn ( 'AsTokenIDs' , lambda x : x . astype ( jnp . int32 ) )