提交 77c42664 authored 作者: Benjamin Scellier's avatar Benjamin Scellier 提交者: Nicolas Ballas

file theano/gof/tests/test_graph.py

上级 2d51062f
...@@ -4,7 +4,7 @@ import pickle ...@@ -4,7 +4,7 @@ import pickle
import unittest import unittest
from nose.plugins.skip import SkipTest from nose.plugins.skip import SkipTest
import numpy import numpy as np
from theano import ( from theano import (
sparse, sparse,
...@@ -362,7 +362,7 @@ class TestAutoName: ...@@ -362,7 +362,7 @@ class TestAutoName:
r1 = tensor.TensorType(dtype='int32', broadcastable=())('myvar') r1 = tensor.TensorType(dtype='int32', broadcastable=())('myvar')
r2 = tensor.TensorVariable(tensor.TensorType(dtype='int32', r2 = tensor.TensorVariable(tensor.TensorType(dtype='int32',
broadcastable=())) broadcastable=()))
r3 = shared(numpy.random.randn(3, 4)) r3 = shared(np.random.randn(3, 4))
assert r1.auto_name == "auto_" + str(autoname_id) assert r1.auto_name == "auto_" + str(autoname_id)
assert r2.auto_name == "auto_" + str(autoname_id + 1) assert r2.auto_name == "auto_" + str(autoname_id + 1)
assert r3.auto_name == "auto_" + str(autoname_id + 2) assert r3.auto_name == "auto_" + str(autoname_id + 2)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论