提交 2c0ec435 authored 作者: Frederic Bastien's avatar Frederic Bastien

flake8

上级 a44e2e50
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
GPU power consumption then gemm call. GPU power consumption then gemm call.
""" """
from __future__ import absolute_import, print_function, division
import numpy as np import numpy as np
...@@ -18,13 +18,11 @@ def burn(): ...@@ -18,13 +18,11 @@ def burn():
sz = 128 sz = 128
img_shp = [sz, sz, sz, sz] img_shp = [sz, sz, sz, sz]
kern_shp = [sz//2, sz, 3, 3] kern_shp = [sz // 2, sz, 3, 3]
out_shp = get_conv_output_shape(img_shp, kern_shp, 'valid', (1, 1)) out_shp = get_conv_output_shape(img_shp, kern_shp, 'valid', (1, 1))
img = T.tensor4('img') img = T.tensor4('img')
kern = T.tensor4('kern') kern = T.tensor4('kern')
out = T.tensor4('out') out = T.tensor4('out')
desc1 = dnn.GpuDnnConvDesc(border_mode='valid', conv_mode='conv')(
kern.shape)
def rand(shp): def rand(shp):
return np.random.rand(*shp).astype(theano.config.floatX) return np.random.rand(*shp).astype(theano.config.floatX)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论