deepy.layers.Concatenate.setup()

Complexity

Conditions

1

Size

Total Lines

5

Duplication

Lines	0
Ratio	0 %

1 Method

Rating	Name	Duplication	Size	Complexity
A	deepy.layers.Concatenate.output()	0	2	1

import theano.tensor as T
from . import NeuralLayer


class Concatenate(NeuralLayer):
    """
    Concatenate two tensors.
    They should have identical dimensions except the last one.
    """

    def __init__(self, axis=2):
        """
        :type layer1: NeuralLayer
        :type layer2: NeuralLayer
        """
        super(Concatenate, self).__init__("concate")
        self.axis = axis
        if axis < 0:
            raise Exception("There are some bugs in T.concatenate, that axis can not lower than 0")

    def prepare(self):
        self.output_dim = sum(self.input_dims)

    def output(self, *xs):
        return T.concatenate(xs, axis=self.axis)

    def test_output(self, *xs):
        return T.concatenate(xs, axis=self.axis)