def compute(self, inputs, outputs):
    theta = inputs[self._cycle_names['theta']]
    A = _compute_A(self.size, theta)
    x = self._inputs_to_vector(inputs)
    y = A.dot(x)
    self._vector_to_outputs(y, outputs)
    outputs[self._cycle_names['theta_out']] = theta
def compute(self, inputs, outputs):
    theta = inputs[self._cycle_names['theta']]
    A = _compute_A(self.size, theta)
    x = self._inputs_to_vector(inputs)
    y = A.dot(x)
    self._vector_to_outputs(y, outputs)
    outputs[self._cycle_names['theta_out']] = theta