TensorFlow
TensorFlow copied to clipboard
fix codes
just touched 'loss_fn', 'grad', codes