提交 599eec95 authored 作者: Benjamin Scellier's avatar Benjamin Scellier

file theano/gpuarray/tests/rnn_support.py

上级 de366285
...@@ -2,7 +2,7 @@ from __future__ import absolute_import, print_function, division ...@@ -2,7 +2,7 @@ from __future__ import absolute_import, print_function, division
import theano import theano
import theano.tensor as T import theano.tensor as T
import numpy import numpy as np
class Model(object): class Model(object):
...@@ -27,9 +27,9 @@ class Model(object): ...@@ -27,9 +27,9 @@ class Model(object):
def uniform(stdev, size): def uniform(stdev, size):
"""uniform distribution with the given stdev and size""" """uniform distribution with the given stdev and size"""
return numpy.random.uniform( return np.random.uniform(
low=-stdev * numpy.sqrt(3), low=-stdev * np.sqrt(3),
high=stdev * numpy.sqrt(3), high=stdev * np.sqrt(3),
size=size size=size
).astype(theano.config.floatX) ).astype(theano.config.floatX)
...@@ -37,7 +37,7 @@ def uniform(stdev, size): ...@@ -37,7 +37,7 @@ def uniform(stdev, size):
def linear_transform_weights(input_dim, output_dim, def linear_transform_weights(input_dim, output_dim,
param_list=None, name=""): param_list=None, name=""):
"theano shared variable given input and output dimension" "theano shared variable given input and output dimension"
weight_inialization = uniform(numpy.sqrt(2.0 / input_dim), weight_inialization = uniform(np.sqrt(2.0 / input_dim),
(input_dim, output_dim)) (input_dim, output_dim))
W = theano.shared(weight_inialization, name=name) W = theano.shared(weight_inialization, name=name)
...@@ -49,7 +49,7 @@ def linear_transform_weights(input_dim, output_dim, ...@@ -49,7 +49,7 @@ def linear_transform_weights(input_dim, output_dim,
def bias_weights(length, param_list=None, name=""): def bias_weights(length, param_list=None, name=""):
"theano shared variable for bias unit, given length" "theano shared variable for bias unit, given length"
bias_initialization = numpy.zeros(length).astype(theano.config.floatX) bias_initialization = np.zeros(length).astype(theano.config.floatX)
bias = theano.shared( bias = theano.shared(
bias_initialization, bias_initialization,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论