class_docstr: Linearly scaled scheduler.
docstr: Get the value for the given step.
docstr_tok: ["Get","the","value","for","the","given","step","."]
code: def get(self, step):
      """Get the value for the given step."""
      if self.num_steps == 0:
          return jnp.full_like(
              step, self.final_value, dtype=jnp.float32
          )
      alpha = jnp.minimum(step / self.num_steps, 1.0)
      return (
          1.0 - alpha
      ) * self.initial_value + alpha * self.final_value
  
imports: import abc
  import collections
  import copy
  import math
  from typing import Any, Iterable, List, Tuple, Union
  from jax import numpy
label_desc: Use image node_id 2 for calling the LinearSchedule obj's underlying member method code with example usage: obj.get(step) and returns: unknown, jnp