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

Make GpuCAReduce interface a little bit more user friendly.

上级 8b22a79b
......@@ -561,6 +561,7 @@ class GpuCAReduce(GpuOp):
self.pre_scalar_op = None
def make_node(self, x):
x = as_cuda_ndarray_variable(x)
if (x.type.ndim != len(self.reduce_mask)):
raise TypeError("x must have rank %i" % len(self.reduce_mask))
o_broadcast = [x.type.broadcastable[i] for i
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论