public override float[] GetPredictedLabels(float threshold = 0f) { Outputs.CopyToHost(); var res = new float[MinibatchSize]; for (var minibatchNo = 0; minibatchNo < MinibatchSize; minibatchNo++) { var maxLabel = float.MinValue; var maxLabelChance = float.MinValue; for (var labelNo = 0; labelNo < this.Size; labelNo++) { var labelChance = Outputs[minibatchNo, labelNo]; if (labelChance > maxLabelChance) { maxLabel = labelNo; maxLabelChance = labelChance; } res[minibatchNo] = maxLabel; } } return(res); }