add comments

Author: frank-dong-ms

src/Microsoft.ML.TensorFlow/TensorflowTransform.cs

@@ -510,13 +510,15 @@ public Mapper(TensorFlowTransformer parent, DataViewSchema inputSchema) :
 
                     if (shape == null || (shape.Length == 0))
                     {
+                        // for vector type input TensorShape should same to dim
                         if (_isInputVector[i])
                         {
                             vecType = (VectorDataViewType)type;
                             var colTypeDims = vecType.Dimensions.Select(dim => (int)dim).ToArray();
                             _fullySpecifiedShapes[i] = new TensorShape(colTypeDims);
                         }
                         else
+                            // for primitive type use default TensorShape
                             _fullySpecifiedShapes[i] = new TensorShape();
                     }
                     else

src/Microsoft.ML.TensorFlow/TensorflowUtils.cs

@@ -81,10 +81,12 @@ internal static DataViewSchema GetModelSchema(IExceptionContext ectx, Graph grap
 
                 if(tensorShape == null)
                 {
+                    // primitive column type
                     schemaBuilder.AddColumn(op.name, mlType, metadataBuilder.ToAnnotations());
                 }
                 else
                 {
+                    // vector column type
                     DataViewType columnType = new VectorDataViewType(mlType);
                     if (!(Utils.Size(tensorShape) == 1 && tensorShape[0] <= 0) &&
                         (Utils.Size(tensorShape) > 0 && tensorShape.Skip(1).All(x => x > 0)))

test/Microsoft.ML.Tests/ScenariosWithDirectInstantiation/TensorflowTests.cs

@@ -1305,8 +1305,10 @@ public void TensorFlowPrimitiveInputTest()
         {
             using var tensorFlowModel = _mlContext.Model.LoadTensorFlowModel(@"model_primitive_input_test");
             var schema = tensorFlowModel.GetModelSchema();
-            Assert.True(schema.TryGetColumnIndex("input1", out var colIndex));
-            Assert.True(schema.TryGetColumnIndex("input2", out colIndex));
+            Assert.True(schema.GetColumnOrNull("input1").HasValue);
+            Assert.True(schema.GetColumnOrNull("input1").Value.Type is TextDataViewType);
+            Assert.True(schema.GetColumnOrNull("input2").HasValue);
+            Assert.True(schema.GetColumnOrNull("input2").Value.Type is TextDataViewType);
 
             var dataview = _mlContext.Data.CreateTextLoader<PrimitiveInput>().Load(new MultiFileSource(null));