def test_3d(self):
exp = np.array([[[0.26726124, 0.53452248, 0.80178373],
[0.26726124, 0.53452248, 0.80178373],
[0.26726124, 0.53452248, 0.80178373]],
[[0.26726124, 0.53452248, 0.80178373],
[0.26726124, 0.53452248, 0.80178373],
[0.26726124, 0.53452248, 0.80178373]]])
got = l2_normalize(np.array([[[1, 2, 3], [1, 2, 3], [1, 2, 3]],
[[1, 2, 3], [1, 2, 3], [1, 2, 3]]]))
np.testing.assert_allclose(exp, got)
def test_3d(self):
exp = np.array([[[0.26726124, 0.53452248, 0.80178373],
[0.26726124, 0.53452248, 0.80178373],
[0.26726124, 0.53452248, 0.80178373]],
[[0.26726124, 0.53452248, 0.80178373],
[0.26726124, 0.53452248, 0.80178373],
[0.26726124, 0.53452248, 0.80178373]]])
got = l2_normalize(
np.array([[[1, 2, 3], [1, 2, 3], [1, 2, 3]],
[[1, 2, 3], [1, 2, 3], [1, 2, 3]]]))
np.testing.assert_allclose(exp, got)