Function gen_matmul_data

imperative/python/test/unit/core/test_custom_op.py:161–168

(seed, m, k, n, low=-0.5, high=0.5, dtype=np.float32)

            return apply(op, ograd, self.lhs, self.rhs)

    def gen_matmul_data(seed, m, k, n, low=-0.5, high=0.5, dtype=np.float32):
        rng = np.random.RandomState(seed=seed)
        lhs_np = rng.uniform(low=low, high=high, size=(m, k)).astype(dtype)
        rhs_np = rng.uniform(low=low, high=high, size=(k, n)).astype(dtype)
        ograd_np = rng.uniform(low=low, high=high, size=(m, n)).astype(dtype)
        scale = rng.uniform(low=0.1, high=0.9, size=(1)).astype(np.float32)[0]

        return lhs_np, rhs_np, ograd_np, scale

    def builtin_func(lhs, rhs, scale):
        out = F.matmul(lhs, rhs) * scale