deepy.layers.Combine

Complexity

Total Complexity

6

Size/Duplication

Total Lines	24
Duplicated Lines	0 %

4 Methods

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

from . import NeuralLayer


class Combine(NeuralLayer):
    """
    Combine two variables.
    """

    def __init__(self, func, dim=0):
        """
        :type layer1: NeuralLayer
        :type layer2: NeuralLayer
        """
        super(Combine, self).__init__("combine")
        self.func = func
        if dim > 0:
            self.output_dim = dim

    def prepare(self):
        if self.output_dim == 0:
            self.output_dim = self.input_dim

    def output(self, *tensors):
        return self.func(*tensors)

    def test_output(self, *tensors):
        return self.func(*tensors)