Softmax3D

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	6	1
A	__init__()	0	2	1

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

from layer import NeuralLayer
import theano
import theano.tensor as T

class Softmax3D(NeuralLayer):

    def __init__(self):
        super(Softmax3D, self).__init__("softmax")

    def compute_tensor(self, x):
        shape = x.shape
        x = x.reshape((-1, shape[-1]))
        softmax_tensor = T.nnet.softmax(x)

        return softmax_tensor.reshape(shape)