test_fill_zeros_like_op.py 403 字节
Newer Older
F
fengjiayi 已提交
1
import unittest
2
from op_test_util import OpTestMeta
F
fengjiayi 已提交
3 4 5 6
import numpy


class TestFillZerosLikeOp(unittest.TestCase):
7
    __metaclass__ = OpTestMeta
F
fengjiayi 已提交
8

9 10
    def setUp(self):
        self.type = "fill_zeros_like"
11 12
        self.inputs = {'Src': numpy.random.random((219, 232)).astype("float32")}
        self.outputs = {'Dst': numpy.zeros_like(self.inputs['Src'])}
F
fengjiayi 已提交
13 14 15 16


if __name__ == '__main__':
    unittest.main()