提交 1105c2bd authored 作者: Shawn Tan's avatar Shawn Tan

Remove duplicated behaviour in child class.

上级 bb3a9a5e
......@@ -1360,11 +1360,6 @@ class GpuExtractDiag(Op):
class GpuAllocDiag(AllocDiag):
__props__ = ("offset", "axis1", "axis2")
def __init__(self, offset=0, axis1=0, axis2=1):
self.offset = offset
self.axis1 = axis1
self.axis2 = axis2
def make_node(self, diag):
ctx_name = infer_context_name(diag)
diag = as_gpuarray_variable(diag, ctx_name)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论