deepy.utils.NeuralTensor

Complexity

Total Complexity

2

Size/Duplication

Total Lines	10
Duplicated Lines	0 %

2 Methods

Rating	Name	Duplication	Size	Complexity
A	wrapper()	0	3	1
A	__getattr__()	0	5	2

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

from theano import tensor
from decorations import neural_computation

class NeuralTensor():
    """
    A class for exporting Theano tensor operations to neural variables.
    """

    def __getattr__(self, func_name):
        @neural_computation
        def wrapper(*args, **kwargs):
            return getattr(tensor, func_name)(*args, **kwargs)
        return wrapper


neural_tensor = NeuralTensor()
NT = neural_tensor