class_docstr: Stochastic Gradient Descent (SGD) optimizer.
docstr: Constructor for SGD optimizer.
  
  Args:
      vc: collection of variables to optimize.
docstr_tok: ["Constructor","for","SGD","optimizer",".","Args",":","vc",":","collection","of","variables","to","optimize","."]
code: def __init__(self, vc: VarCollection):
      """Constructor for SGD optimizer.
  
      Args:
          vc: collection of variables to optimize.
      """
      self.train_vars = ModuleList(
          TrainRef(x) for x in vc.subset(TrainVar)
      )
  
file_path: objax/objax/optimizer/sgd.py
imports: from typing import List
  from objax.module import Module, ModuleList
  from objax.typing import JaxArray
  from objax.util import class_name
  from objax.variable import TrainRef, TrainVar, VarCollection
label_desc: Use image node_id 1 to create a new SGD object from inherited base classes: Module with example: obj = SGD(vc)