def test_power_operators(): a = 2 * nd.ones(shape=LARGE_X) # sqrt result = nd.sqrt(a) assert result[-1] == 1.4142135 assert result.shape == a.shape # rsqrt result = nd.rsqrt(a) assert result[-1] == 0.70710677 assert result.shape == a.shape # cbrt result = nd.cbrt(a) assert result[-1] == 1.2599211 assert result.shape == a.shape # rcbrt result = nd.rcbrt(a) assert result[-1] == 0.7937005 assert result.shape == a.shape # square result = nd.square(a) assert result[-1] == 4 assert result.shape == a.shape # reciprocal result = nd.reciprocal(a) assert result[-1] == 0.5 assert result.shape == a.shape
def rcbrt(x): return nd.rcbrt(x)