There was an error while loading. Please reload this page.
1 parent b29b60d commit 63dabbcCopy full SHA for 63dabbc
src/Microsoft.ML.TensorFlow/TensorflowUtils.cs
@@ -90,7 +90,7 @@ internal static DataViewSchema GetModelSchema(IExceptionContext ectx, Graph grap
90
DataViewType columnType = new VectorDataViewType(mlType);
91
if (!(Utils.Size(tensorShape) == 1 && tensorShape[0] <= 0) &&
92
(Utils.Size(tensorShape) > 0 && tensorShape.Skip(1).All(x => x > 0)))
93
- // When treatOutputAsBatched is true we keep the existing behaviour. This means that if the first dimension is greater
+ // treatOutputAsBatched == true means that if the first dimension is greater
94
// than 0 we take the tensor shape as is. If the first value is less then 0, we treat it as the batch input so we can
95
// ignore it for the shape of the ML.NET vector. I.E. if the input dimensions are [-1, 5], ML.NET will read the -1 as
96
// batch input, and so the ML.NET data type will be a vector of length 5.
0 commit comments