コード例 #1
0
        /// <summary>
        /// Adds an input variable to the list.
        /// </summary>
        public void AddInputVariable(DataViewType type, string colName)
        {
            _host.CheckValue(type, nameof(type));
            _host.CheckValue(colName, nameof(colName));

            colName = AddVariable(colName);
            _inputs.Add(OnnxUtils.GetModelArgs(type, colName));
        }
コード例 #2
0
 /// <summary>
 /// Adds an intermediate column to the list.
 /// </summary>
 public override string AddIntermediateVariable(DataViewType type, string colName, bool skip = false)
 {
     colName = AddVariable(colName);
     // Let the runtime figure the shape.
     if (!skip)
     {
         _host.CheckValue(type, nameof(type));
         _intermediateValues.Add(OnnxUtils.GetModelArgs(type, colName));
     }
     return(colName);
 }
コード例 #3
0
 /// <summary>
 /// Adds an output variable to the list.
 /// </summary>
 public void AddOutputVariable(DataViewType type, string variableName, List <long> dim = null)
 {
     _host.CheckValue(type, nameof(type));
     _host.CheckParam(IsVariableDefined(variableName), nameof(variableName));
     _outputs.Add(OnnxUtils.GetModelArgs(type, variableName, dim));
 }