load_state_dict(state_dict)
  Loads the scaler state.  If this instance is disabled,
  "load_state_dict()" is a no-op.

  Parameters:
     **state_dict** (*dict*) -- scaler state.  Should be an object
     returned from a call to "state_dict()".

scale(outputs)
  Multiplies ('scales') a tensor or list of tensors by the scale
  factor.

  Returns scaled outputs.  If this instance of "GradScaler" is not
  enabled, outputs are returned unmodified.

  Parameters:
     **outputs** (*Tensor** or **iterable of Tensors*) -- Outputs
     to scale.

set_backoff_factor(new_factor)
  Parameters:
     **new_scale** (*float*) -- Value to use as the new scale
     backoff factor.

set_growth_factor(new_factor)
  Parameters:
     **new_scale** (*float*) -- Value to use as the new scale
     growth factor.

set_growth_interval(new_interval)
  Parameters:
     **new_interval** (*int*) -- Value to use as the new growth
