提交 86c150d8 authored 作者: hantek's avatar hantek

Merge branch 'allocdiag' of https://github.com/hantek/Theano into allocdiag

......@@ -1201,9 +1201,14 @@ class GpuAllocDiag(Subtensor):
def perform(self, node, inputs, outputs):
(x,) = inputs
(z,) = outputs
pdb.set_trace()
z = alloc zero mtx in cudandarray
z = set subtensor
dim = x.shape[0] + self.offset
z = gpuarray.zeros((dim, dim))
# z.subtensor
# alloc zero mtx in cudandarray
# z = set subtensor
def grad(self, inputs, gout):
(input_x,) = inputs
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论