Combine

Complexity

Total Complexity

6

Size/Duplication

Total Lines	24
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	__init__()	0	9	2
A	prepare()	0	3	2
A	compute_tensor()	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 compute_tensor(self, *tensors):
        return self.func(*tensors)

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