Python Splitの例

コード例 #1

ファイル: basic_ops.py プロジェクト: 5730279821-TA/Theano

 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

 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)