public void SaveAsOnnx(OnnxContext ctx)
            {
                var droppedCols = new HashSet <int>(Enumerable.Range(0, InputSchema.Count));

                var outputToInputMap = _mapper.OutputToInputMap;

                for (int i = 0; i < outputToInputMap.Length; i++)
                {
                    var    srcCol      = InputSchema[outputToInputMap[i]];
                    var    dstCol      = OutputSchema[i];
                    var    srcVariable = ctx.GetVariableName(srcCol.Name);
                    var    dstVariable = ctx.AddIntermediateVariable(dstCol.Type, dstCol.Name, true);
                    string opType      = "Identity";
                    ctx.CreateNode(opType, srcVariable, dstVariable, ctx.GetNodeName(opType), "");

                    droppedCols.Remove(srcCol.Index);
                }

                foreach (var srcCol in droppedCols)
                {
                    ctx.RemoveInputVariable(InputSchema[srcCol].Name);
                }
            }