deepy.layers.Flatten

Complexity

Total Complexity

2

Size/Duplication

Total Lines	10
Duplicated Lines	0 %

2 Methods

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

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

from . import NeuralLayer
import theano.tensor as T

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

    def __init__(self):
        super(Flatten, self).__init__("flatten")
    
    def output(self, x):
        return T.flatten(x, 2)