def _test_repeat(): print '\n------------------------------------------------------------' print 'Test: dlx.util.theano_utils.repeat' x2 = TU.tensor(2, 'x2') x3 = TU.tensor(3, 'x3') y3 = TU.repeat(x2, 10) y4 = TU.repeat(x3, 10) print P.pprint(y3) print P.pprint(y4)
def output(self, train=False): X = self.get_input('input')(train) return TU.repeat(X, self.n).dimshuffle((1, 0, 2))