def setUp(self): class Model(chainer.Chain): def __init__(self, ops): super(Model, self).__init__() self.ops = ops def __call__(self, a, b): if not isinstance(a, chainer.Variable): a = chainer.Variable(a) if not isinstance(b, chainer.Variable): b = chainer.Variable(b) return eval(self.ops) self.model = Model(self.ops) a = chainer.Variable(input_generator.increasing(2, 3)) b = chainer.Variable(input_generator.nonzero_increasing(2, 3) * 0.3) self.x = (a, b)
def vector(self): a = chainer.Variable(input_generator.positive_increasing(2, )) b = chainer.Variable(input_generator.nonzero_increasing(2, ) * 0.3) return (a, b)
def matrix(self): a = chainer.Variable(input_generator.positive_increasing(5, 2, 3)) b = chainer.Variable(input_generator.nonzero_increasing(5, 2, 3) * 0.3) return (a, b)