DimShuffle

Complexity

Total Complexity

4

Size/Duplication

Total Lines	14
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__init__()	0	6	3
A	compute_tensor()	0	2	1

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from . import NeuralLayer

class DimShuffle(NeuralLayer):
    """
    DimShuffle layer.
    """

    def __init__(self, *pattern):
        super(DimShuffle, self).__init__("dimshuffle")
        if len(pattern) == 1 and type(pattern[0]) == list:
            self.pattern = pattern[0]
        else:
            self.pattern = pattern

    def compute_tensor(self, x):
        return x.dimshuffle(*self.pattern)