deepy.layers.NeuralVariable

Complexity

Total Complexity

11

Size/Duplication

Total Lines	42
Duplicated Lines	0 %

9 Methods

Rating	Name	Duplication	Size	Complexity
A	compute_tensor()	0	2	1
A	apply()	0	6	2
A	dim()	0	2	1
A	set_test_value()	0	2	1
A	__init__()	0	9	2
A	shape()	0	2	1
A	getitem_wrapper()	0	3	1
A	compute_test_tesnor()	0	2	1
A	__getitem__()	0	5	2

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

from layer import NeuralLayer
from deepy.utils.decorations import neural_computation


class NeuralVariable(NeuralLayer):
    """
    Create a constant layer with tensors.
    """

    def __init__(self, tensor, test_tensor=None, dim=0):
        """
        Create a tensor layer.
        """
        super(NeuralVariable, self).__init__("const")
        self.output_dim = dim
        self.tensor = tensor
        self.test_tensor = tensor if not test_tensor else test_tensor
        self.initialize(0)

    def __getitem__(self, index):
        @neural_computation
        def getitem_wrapper(t, index):
            return t[index]
        return getitem_wrapper(self, index)

    def apply(self, func, dim=None):
        """
        Apply a function to tensors.
        """
        output_dim = dim if dim else self.output_dim
        return NeuralVariable(func(self.tensor), func(self.test_tensor), output_dim)

    def compute_tensor(self, x):
        return self.tensor

    def compute_test_tesnor(self, x):
        return self.test_tensor

    def set_test_value(self, value):
        self.tensor.tag.test_value = value

    def dim(self):
        return self.output_dim

    def shape(self, dim_index):
        return self.tensor.shape[dim_index]