Python Join.make_nodeの例

コード例 #1

    def make_node(self, axis, *tensors):
        node = Join.make_node(self, axis, *tensors)

        return Apply(self, [node.inputs[0]] + map(as_gpuarray_variable,
                                                  tensors),
                     [GpuArrayType(broadcastable=node.outputs[0].broadcastable,
                                   dtype=node.outputs[0].dtype)()])

コード例 #2

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

    def make_node(self, axis, *tensors):
        node = Join.make_node(self, axis, *tensors)

        return Apply(self, [node.inputs[0]] + list(map(as_gpuarray_variable,
                                                  tensors)),
                     [GpuArrayType(broadcastable=node.outputs[0].broadcastable,
                                   dtype=node.outputs[0].dtype)()])

コード例 #3

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

    def make_node(self, axis, *tensors):
        node = Join.make_node(self, axis, *tensors)

        ctx_name = infer_context_name(*tensors)

        def agv(v):
            return as_gpuarray_variable(v, context_name=ctx_name)

        return Apply(self, [node.inputs[0]] + list(map(agv, tensors)),
                     [GpuArrayType(broadcastable=node.outputs[0].broadcastable,
                                   dtype=node.outputs[0].dtype,
                                   context_name=ctx_name)()])

コード例 #4

    def make_node(self, axis, *tensors):
        node = Join.make_node(self, axis, *tensors)

        ctx_name = infer_context_name(*tensors)

        def agv(v):
            return as_gpuarray_variable(v, context_name=ctx_name)

        return Apply(self, [node.inputs[0]] + list(map(agv, tensors)),
                     [GpuArrayType(broadcastable=node.outputs[0].broadcastable,
                                   dtype=node.outputs[0].dtype,
                                   context_name=ctx_name)()])

コード例 #5

ファイル: mkl_elemwise.py プロジェクト: intel/theano

    def make_node(self, *tensors):
        # Neet to check ndim and shape of all input tensors!
        for x in tensors:
            assert x.type.ndim == 4

        node = Join.make_node(self, 1, *tensors)

        def agv(v):
            return as_tensor_variable(v)
        return Apply(self, list(map(agv, tensors)),
                     [TensorType(dtype=node.outputs[0].dtype,
                      broadcastable=node.outputs[0].broadcastable)()])

コード例 #6

 def make_node(self, *axis_and_tensors):
     return Join.make_node(self, *axis_and_tensors)

コード例 #7

ファイル: mkl_concatenate.py プロジェクト: pcs-theano/Theano

 def make_node(self, *axis_and_tensors):
     return Join.make_node(self, *axis_and_tensors)