public void cntk_argmax_test()
{
using (var K = new CNTKBackend())
{
var x = K.variable(array: new double[, ] {
{ 1, 9, 3 }, { 9, 5, 6 }
});
Assert.AreEqual(new double[] { 1, 0, 1 }, K.argmax(x, axis: 0).eval());
Assert.AreEqual(new double[] { 1, 0 }, K.argmax(x, axis: 1).eval());
Assert.AreEqual(new double[] { 1, 0 }, K.argmax(x).eval());
}
}