Update code example to use the context argument.

rxwei · web-flow · commit 8bb9b1a4e794 · 2019-02-21T03:03:47.000-08:00
diff --git a/README.md b/README.md
@@ -29,10 +29,10 @@ struct Model: Layer {
     var layer3 = Dense<Float>(inputSize: hiddenSize, outputSize: 3, activation: identity)
     
     @differentiable(wrt: (self, input))
-    func applied(to input: Tensor<Float>) -> Tensor<Float> {
-        let l1 = layer1.applied(to: input)
-        let l2 = layer2.applied(to: l1)
-        return layer3.applied(to: l2)
+    func applied(to input: Tensor<Float>, in context: Context) -> Tensor<Float> {
+        let l1 = layer1.applied(to: input, in: context)
+        let l2 = layer2.applied(to: l1, in: context)
+        return layer3.applied(to: l2, in: context)
     }
 }
 ```

Original file line number	Diff line number	Diff line change
`@@ -29,10 +29,10 @@ struct Model: Layer {`
`29`	`29`	`var layer3 = Dense<Float>(inputSize: hiddenSize, outputSize: 3, activation: identity)`
`30`	`30`
`31`	`31`	`@differentiable(wrt: (self, input))`
`32`		`- func applied(to input: Tensor<Float>) -> Tensor<Float> {`
`33`		`- let l1 = layer1.applied(to: input)`
`34`		`- let l2 = layer2.applied(to: l1)`
`35`		`- return layer3.applied(to: l2)`
	`32`	`+ func applied(to input: Tensor<Float>, in context: Context) -> Tensor<Float> {`
	`33`	`+ let l1 = layer1.applied(to: input, in: context)`
	`34`	`+ let l2 = layer2.applied(to: l1, in: context)`
	`35`	`+ return layer3.applied(to: l2, in: context)`
`36`	`36`	`}`
`37`	`37`	`}`
`38`	`38`	```