Exemplo n.º 1
0
 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());
     }
 }