for testing and deploying your application
for finding and fixing issues
for empowering human code reviews
#!/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]
uaca /
deepy
