提交 a9530794 authored 作者: Frederic's avatar Frederic

reuse random value

上级 7b619d38
...@@ -21,19 +21,14 @@ def main(dev1, dev2): ...@@ -21,19 +21,14 @@ def main(dev1, dev2):
init_dev(dev2, 'ctx2') init_dev(dev2, 'ctx2')
size = 1024 * 16 size = 1024 * 16
val1a = shared(numpy.random.randn(size, size).astype('float32'), data = numpy.random.randn(size, size).astype('float32')
target='ctx1') val1a = shared(data, target='ctx1')
val1b = shared(numpy.random.randn(size, size).astype('float32'), val1b = shared(data, target='ctx1')
target='ctx1') val1c = shared(data, target='ctx1')
val1c = shared(numpy.random.randn(size, size).astype('float32'), val1d = shared(data, target='ctx1')
target='ctx1')
val1d = shared(numpy.random.randn(size, size).astype('float32'), val2a = shared(data, target='ctx2')
target='ctx1') val2b = shared(data, target='ctx2')
val2a = shared(numpy.random.randn(size, size).astype('float32'),
target='ctx2')
val2b = shared(numpy.random.randn(size, size).astype('float32'),
target='ctx2')
f1 = theano.function([], [gpu_dot22(val1a, val1b), f1 = theano.function([], [gpu_dot22(val1a, val1b),
gpu_dot22(val1c, val1d)]) gpu_dot22(val1c, val1d)])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论