deepy.layers.Concatenate

Complexity

Total Complexity

5

Size/Duplication

Total Lines	24
Duplicated Lines	0 %

4 Methods

Rating	Name	Duplication	Size	Complexity
A	test_output()	0	2	1
A	output()	0	2	1
A	prepare()	0	2	1
A	__init__()	0	9	2

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)