def test_rff_feature_map_derivatives2_equals_feature_map_derivatives2_loop(): N = 10 D = 20 m = 3 X = np.random.randn(N, D) omega = np.random.randn(D, m) u = np.random.uniform(0, 2 * np.pi, m) derivatives = rff_feature_map_grad2(X, omega, u) derivatives_loop = rff_feature_map_grad2_loop(X, omega, u) assert_allclose(derivatives_loop, derivatives)
def compute_b_memory(X, omega, u): assert len(X.shape) == 2 Phi2 = rff_feature_map_grad2(X, omega, u) return -np.mean(np.sum(Phi2, 0), 0)