deepy.layers.Combine

Complexity

Total Complexity

6

Size/Duplication

Total Lines	24
Duplicated Lines	0 %

4 Methods

Rating	Name	Duplication	Size	Complexity
A	Combine.compute_test_tesnor()	0	2	1
A	Combine.compute_tensor()	0	2	1
A	Combine.__init__()	0	9	2
A	Combine.prepare()	0	3	2

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 compute_tensor(self, *tensors):
        return self.func(*tensors)

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