deepy.layers.NeuralVar

Complexity

Total Complexity

9

Size/Duplication

Total Lines	36
Duplicated Lines	0 %

7 Methods

Rating	Name	Duplication	Size	Complexity
A	test_output()	0	2	1
A	dim()	0	2	1
A	__init__()	0	9	2
A	output()	0	2	1
A	apply()	0	6	2
A	set_test_value()	0	2	1
A	shape()	0	2	1

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

from layer import NeuralLayer


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

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

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

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

    def test_output(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]