def setUp(self):
f = T.Variable(sum, 'sum')
self.c1 = T.Call(f, ([1, 2, 3], ))
self.c2 = f([1, 2, 3])
self.c3 = T.Call(f, (['', 1], ))
def setUp(self):
x = T.Variable(1, 'x')
f = lambda x: (x, )
self.op1 = T.UnaryOperator(8, x, '-', f)
self.op2 = T.UnaryOperator(9, x, '-', f)
def setUp(self):
x = Object()
self.value = T.GetAttr(T.Variable(x, 'x'), 'value')
self.value2 = T.GetAttr(T.Variable(x, 'x'), T.Constant('value'))
self.value3 = T.GetAttr(T.Variable(x, 'x'), 3)
def setUp(self):
x = T.Variable([1, 2, 3], 'x')
y = T.Variable({'a': 1, 'b': 2}, 'y')
self.x = x
self.v1 = T.GetItem(x, 1)
self.v2 = T.GetItem(y, 'a')
def setUp(self):
self.x = T.Variable(1, 'x')
self.y = T.Variable(2, 'y')
def setUp(self):
self.x = T.Variable(10, 'x')
def make_multiple_tc_variable(ty_args, names):
assert len(ty_args) == len(names)
return [type_check.Variable(t, n) for t, n in zip(ty_args, names)]
def apply_subst_shapeElem(subst, e):
if e.value in subst.keys():
return ShapeElem(subst[e.value],
expr=type_check.Variable(None, e.value))
return e
import numpy
from chainer import function
from chainer.utils import type_check
_type_check_prod = type_check.Variable(numpy.prod, 'prod')
class Reshape(function.Function):
"""Reshapes an input array without copy."""
def __init__(self, shape):
self.shape = shape
def check_type_forward(self, in_types):
type_check.expect(
in_types.size() == 1,
_type_check_prod(in_types[0].shape) == _type_check_prod(
self.shape))
def forward(self, x):
return x[0].reshape(self.shape),
def backward(self, x, gy):
return gy[0].reshape(x[0].shape),
def reshape(x, shape):
"""Reshapes an input variable without copy.
Args:
x (~chainer.Variable): Input variable.
