Method forward

tests/python/nightly/test_nnapi/test_from_exported_to_cuda.py:722–723

(self, x)

            self.y = torch.tensor(np.random.rand(2, 3).astype("float32"))

        def forward(self, x):
            return x.mul(self.y)

    torch_module = MulModule().eval()
    raw_data = np.random.rand(2, 3).astype("float32")