The `assign` operation will record gradient. The logic is missing in TensorFlow.NET. 