Skip to content

Instantly share code, notes, and snippets.

Created September 28, 2021 16:32
What would you like to do?
// Create an array feature
var logits = new MLArrayFeature<float>(...); // shape: (1, 1000)
// Perform argmax
var result = logits.ArgMax(dim: 1).Item;
var resultLabel = labels[result];
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment