commit
3a8017d148
@ -1,2 +1,2 @@
|
||||
cc_library(paddle_pybind SHARED SRCS pybind.cc DEPS pybind python
|
||||
add_op fc_op sgd_op cross_entropy_op recurrent_network_op)
|
||||
add_op fc_op sgd_op cross_entropy_op recurrent_network_op fill_zeros_like_op)
|
||||
|
@ -0,0 +1,35 @@
|
||||
import unittest
|
||||
import paddle.v2.framework.create_op_creation_methods as creation
|
||||
import paddle.v2.framework.core as core
|
||||
from op_test_util import OpTestMeta
|
||||
import numpy
|
||||
|
||||
|
||||
class TestFillZerosLikeOp(unittest.TestCase):
|
||||
def test_fill(self):
|
||||
scope = core.Scope(None)
|
||||
a = scope.create_var("input")
|
||||
a_tensor = a.get_tensor()
|
||||
a_tensor.set_dims([546, 291])
|
||||
a_tensor.alloc_float()
|
||||
a_tensor.set(numpy.random.random((546, 291)).astype("float32"))
|
||||
|
||||
op = creation.op_creations.fill_zeros_like(Src="input", Dst="output")
|
||||
|
||||
for out in op.outputs():
|
||||
if scope.get_var(out) is None:
|
||||
scope.create_var(out).get_tensor()
|
||||
|
||||
b_tensor = scope.get_var("output").get_tensor()
|
||||
op.infer_shape(scope)
|
||||
self.assertEqual([546, 291], b_tensor.shape())
|
||||
ctx = core.DeviceContext.cpu_context()
|
||||
op.run(scope, ctx)
|
||||
b_tensor_array = numpy.array(b_tensor)
|
||||
for r in range(0, 546):
|
||||
for c in range(0, 291):
|
||||
self.assertEqual(b_tensor_array[r][c], 0.0)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
Loading…
Reference in new issue