Пример #1
0
 def make_node(self, x, axis, splits):
     node = Split.make_node(self, x, axis, splits)
     x = as_gpuarray_variable(x, infer_context_name(x))
     outs = [GpuArrayType(dtype=o.dtype, broadcastable=o.broadcastable,
                          context_name=x.type.context_name)()
             for o in node.outputs]
     return Apply(self, [x] + node.inputs[1:], outs)
Пример #2
0
 def make_node(self, x, axis, splits):
     node = Split.make_node(self, x, axis, splits)
     x = as_gpuarray_variable(x, infer_context_name(x))
     outs = [GpuArrayType(dtype=o.dtype, broadcastable=o.broadcastable,
                          context_name=x.type.context_name)()
             for o in node.outputs]
     return Apply(self, [x] + node.inputs[1:], outs)