@Test
public void run() {
try (Graph g = new Graph();
Session s = new Session(g)) {
TestUtil.transpose_A_times_X(g, new int[][] {{2}, {3}});
try (Tensor x = Tensor.create(new int[][] {{5}, {7}});
AutoCloseableList<Tensor> outputs =
new AutoCloseableList<Tensor>(s.runner().feed("X", x).fetch("Y").run())) {
assertEquals(1, outputs.size());
final int[][] expected = {{31}};
assertEquals(expected, outputs.get(0).copyTo(new int[1][1]));
}
}
}
