fixed the false-branch of the switch graph

henon · henon · commit 40dccd362b7d · 2019-04-11T12:53:41.000+02:00
diff --git a/src/TensorFlowNET.Core/Operations/ControlFlows/CondContext.cs b/src/TensorFlowNET.Core/Operations/ControlFlows/CondContext.cs
@@ -207,6 +207,9 @@ private Tensor _ProcessOutputTensor(Tensor val)
                     _values.Add(real_val.name);
                     _external_values[real_val.name] = real_val;
                 }
+                var (t0, t1) = control_flow_ops._SwitchRefOrTensor(real_val, _pred);
+                real_val = new[] {t0, t1}[_branch];
+                _external_values[val.name] = real_val;
             }
             else
             {

Original file line number	Diff line number	Diff line change
`@@ -207,6 +207,9 @@ private Tensor _ProcessOutputTensor(Tensor val)`
`207`	`207`	`_values.Add(real_val.name);`
`208`	`208`	`_external_values[real_val.name] = real_val;`
`209`	`209`	`}`
	`210`	`+ var (t0, t1) = control_flow_ops._SwitchRefOrTensor(real_val, _pred);`
	`211`	`+ real_val = new[] {t0, t1}[_branch];`
	`212`	`+ _external_values[val.name] = real_val;`
`210`	`213`	`}`
`211`	`214`	`else`
`212`	`215`	`{`