module onnxrt.ops_cpu.op_gather_#

Short summary#

module mlprodict.onnxrt.ops_cpu.op_gather_

Implements runtime for operator Gather. The code is inspired from tfidfvectorizer.cc in onnxruntime.

source on GitHub

Classes#

class

truncated documentation

GatherDouble

Implements runtime for operator Gather. The code is inspired from tfidfvectorizer.cc …

GatherFloat

Implements runtime for operator Gather. The code is inspired from tfidfvectorizer.cc …

GatherInt64

Implements runtime for operator Gather. The code is inspired from tfidfvectorizer.cc …

Documentation#

Implements runtime for operator Gather. The code is inspired from tfidfvectorizer.cc in onnxruntime.

class mlprodict.onnxrt.ops_cpu.op_gather_.GatherDouble(self: mlprodict.onnxrt.ops_cpu.op_gather_.GatherDouble, arg0: int)#

Bases: pybind11_object

Implements runtime for operator Gather. The code is inspired from tfidfvectorizer.cc in onnxruntime.

__init__(self: mlprodict.onnxrt.ops_cpu.op_gather_.GatherDouble, arg0: int) → None#
compute(self: mlprodict.onnxrt.ops_cpu.op_gather_.GatherDouble, arg0: numpy.ndarray[numpy.float64], arg1: numpy.ndarray[numpy.int64]) → numpy.ndarray[numpy.float64]#

Computes Gather.

class mlprodict.onnxrt.ops_cpu.op_gather_.GatherFloat(self: mlprodict.onnxrt.ops_cpu.op_gather_.GatherFloat, arg0: int)#

Bases: pybind11_object

Implements runtime for operator Gather. The code is inspired from tfidfvectorizer.cc in onnxruntime.

__init__(self: mlprodict.onnxrt.ops_cpu.op_gather_.GatherFloat, arg0: int) → None#
compute(self: mlprodict.onnxrt.ops_cpu.op_gather_.GatherFloat, arg0: numpy.ndarray[numpy.float32], arg1: numpy.ndarray[numpy.int64]) → numpy.ndarray[numpy.float32]#

Computes Gather.

class mlprodict.onnxrt.ops_cpu.op_gather_.GatherInt64(self: mlprodict.onnxrt.ops_cpu.op_gather_.GatherInt64, arg0: int)#

Bases: pybind11_object

Implements runtime for operator Gather. The code is inspired from tfidfvectorizer.cc in onnxruntime.

__init__(self: mlprodict.onnxrt.ops_cpu.op_gather_.GatherInt64, arg0: int) → None#
compute(self: mlprodict.onnxrt.ops_cpu.op_gather_.GatherInt64, arg0: numpy.ndarray[numpy.int64], arg1: numpy.ndarray[numpy.int64]) → numpy.ndarray[numpy.int64]#

Computes Gather.

mlprodict.onnxrt.ops_cpu.op_gather_.InferShapeGather(arg0: List[int], arg1: List[int], arg2: int) → List[int]#

Infer shapes for Gather operators.