Пример #1
0
 public void AddConstant(string name, ONNXTensor onnxTensor)
 {
     if (!onnxTensor.IsEmpty())
     {
         constants[name] = onnxTensor;
         AddVariable(name, onnxTensor);
     }
 }
Пример #2
0
 public void AddVariable(string nodeId, ONNXTensor onnxTensor)
 {
     variables[nodeId] = new VariableTensor {
         features       = -1,
         rank           = onnxTensor.rank,
         layout         = VariableTensor.Layout.Unknown,
         productOfShape = null
     };
 }
Пример #3
0
 public ONNXTensor GetOptionalTensor(string name, ONNXTensor defaultValue)
 {
     try { return(GetRequiredTensor(name)); }
     catch (OnnxLayerImportException) { return(defaultValue); }
 }