module onnxrt.ops_cpu.op_gathernd

Short summary

module mlprodict.onnxrt.ops_cpu.op_gathernd

Runtime operator.

source on GitHub

Classes

class	truncated documentation
GatherND	Python runtime for function SoftmaxCrossEntropyLoss.

Functions

function	truncated documentation
_gather_nd_impl	Modified version of softmaxcrossentropy.py. …

Properties

property	truncated documentation
args_default	Returns the list of arguments as well as the list of parameters with the default values (close to the signature). …
args_default_modified	Returns the list of modified parameters.
args_mandatory	Returns the list of optional arguments.
args_optional	Returns the list of optional arguments.
atts_value	Returns all parameters in a dictionary.

Methods

method	truncated documentation
__init__
_run

Documentation

Runtime operator.

source on GitHub

class mlprodict.onnxrt.ops_cpu.op_gathernd.GatherND(onnx_node, desc=None, **options)

Bases: OpRun

Python runtime for function SoftmaxCrossEntropyLoss.

source on GitHub

__init__(onnx_node, desc=None, **options)

_run(data, indices, attributes=None, verbose=0, fLOG=None)

Should be overwritten.

source on GitHub

mlprodict.onnxrt.ops_cpu.op_gathernd._gather_nd_impl(data, indices, batch_dims)

Modified version of softmaxcrossentropy.py.

source on GitHub