Apply One Full SGD Training Step | TensorTonic