def test_argmax_axis1(self): be = CPU() tsr = be.array([[-1, 10], [11, 9]]) out = be.empty((2, )) be.argmax(tsr, 1, out) assert_tensor_equal(out, CPUTensor([1, 0]))
def test_argmax_axis1(self): be = CPU() tsr = be.array([[-1, 10], [11, 9]]) out = be.empty((2, )) be.argmax(tsr, 1, out) assert_tensor_equal(out, CPUTensor([1, 0]))
def test_argmax_noaxis(self): be = CPU() tsr = be.array([[-1, 0], [1, 92]]) out = be.empty([1, 1]) be.argmax(tsr, None, out) assert_tensor_equal(out, CPUTensor(3))
def test_argmax_noaxis(self): be = CPU() tsr = be.array([[-1, 0], [1, 92]]) out = be.empty([1, 1]) be.argmax(tsr, None, out) assert_tensor_equal(out, CPUTensor(3))