提交 7b96553f authored 作者: Zhouhan LIN's avatar Zhouhan LIN

minor changes

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