test_gather_op.py

import unittest

import numpy
import paddle.v2.framework.core as core
from paddle.v2.framework.op import Operator

from op_test_util import OpTestMeta


class TestGatherOp(unittest.TestCase):
    __metaclass__ = OpTestMeta

    def setUp(self):
        self.type = "gather"
        xnp = numpy.random.random((10, 20)).astype("float32")
        self.inputs = {
            'X': xnp,
            'Index': numpy.array([1, 3, 5]).astype("int32")
        }
        self.outputs = {'Y': self.inputs['X'][self.inputs['Index']]}


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