18 lines
406 B
18 lines
406 B
import unittest
|
|
from op_test_util import OpTestMeta
|
|
import numpy as np
|
|
|
|
|
|
class TestMulOp(unittest.TestCase):
|
|
__metaclass__ = OpTestMeta
|
|
|
|
def setUp(self):
|
|
self.type = "mul"
|
|
self.X = np.random.random((32, 784)).astype("float32")
|
|
self.Y = np.random.random((784, 100)).astype("float32")
|
|
self.Out = np.dot(self.X, self.Y)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|