def test_axis(self):
t = TensorBase(
np.array([[0.77937768, 0.51022484, 0.49155195, 0.02769902],
[0.03777148, 0.13020167, 0.02155692, 0.69574893]]))
result = t.max(axis=1)
self.assertTrue(syft.equal(result, [0.77937768, 0.69574893]))
result = t.max(axis=0)
self.assertTrue(
syft.equal(result,
[0.77937768, 0.51022484, 0.49155195, 0.69574893]))
def testNoDim(self):
t = TensorBase(np.array([[0.77937768, 0.51022484, 0.49155195, 0.02769902], [0.03777148, 0.13020167, 0.02155692, 0.69574893]]))
self.assertTrue(t.max() == 0.77937768)