Reshape

Complexity

Total Complexity

2

Size/Duplication

Total Lines	11
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

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

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

from . import NeuralLayer

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

    def __init__(self, pattern):
        super(Reshape, self).__init__("dimshuffle")
        self.pattern = pattern

    def compute_tensor(self, x):
        return x.reshape(self.pattern)