def setUp(self): y_shape = self.y_shape x_shape = self.x_shape y1 = make_array(1, y_shape, numpy.float32) y2 = make_array(2, y_shape, numpy.float32) gx1 = make_array(1, x_shape, numpy.float32) gx2 = None gy1 = make_array(1, y_shape, numpy.float32) gy2 = make_array(1, y_shape, numpy.float32) f = chainer.Function() f.check_type_forward = mock.MagicMock() f.forward_cpu = mock.MagicMock(return_value=(y1, y2)) f.forward_gpu = mock.MagicMock() f.backward_cpu = mock.MagicMock(return_value=(gx1, gx2)) f.backward_gpu = mock.MagicMock() self.f = f self.x1 = make_array(0, x_shape, numpy.float32) self.x2 = make_array(0, x_shape, numpy.int32) self.y1 = y1 self.y2 = y2 self.gx1 = gx1 self.gx2 = gx2 self.gy1 = gy1 self.gy2 = gy2
def test_set_fresh_creator(self): v = chainer.Variable() f = chainer.Function() v.creator = f self.assertIs(v.creator, f) self.assertIs(v.creator_node, f.node) self.assertEqual(v.rank, 1)
def setUp(self): f = chainer.Function() f.p1 = numpy.arange(10) f.p2 = numpy.arange(5) f.g1 = numpy.arange(8) f.g2 = numpy.arange(3) f.parameter_names = ('p1', 'p2') f.gradient_names = ('g1', 'g2') self.f = f
def setUp(self): y1 = numpy.arange(4).astype(numpy.float32) y2 = numpy.arange(4).astype(numpy.float32) + 1 gx1 = numpy.arange(3).astype(numpy.float32) gx2 = None gy1 = numpy.arange(4).astype(numpy.float32) gy2 = numpy.arange(4).astype(numpy.float32) f = chainer.Function() f.check_type_forward = mock.MagicMock() f.forward_cpu = mock.MagicMock(return_value=(y1, y2)) f.forward_gpu = mock.MagicMock() f.backward_cpu = mock.MagicMock(return_value=(gx1, gx2)) f.backward_gpu = mock.MagicMock() self.f = f self.x1 = numpy.arange(3).astype(numpy.float32) self.x2 = numpy.arange(3).astype(numpy.int32) self.y1 = y1 self.y2 = y2 self.gx1 = gx1 self.gx2 = gx2 self.gy1 = gy1 self.gy2 = gy2
def setUp(self): self.original_debug = chainer.is_debug() chainer.set_debug(True) self.one = numpy.array([1], numpy.float32) self.f = chainer.Function()
def test_label(self): self.assertEqual(chainer.Function().label, 'Function')
def test_label(self): self.assertEqual(chainer.Function().label, '<class \'chainer.function.Function\'>')