module onnxrt.ops_cpu.op_softmax_cross_entropy_loss

Short summary

module mlprodict.onnxrt.ops_cpu.op_softmax_cross_entropy_loss

Runtime operator.

source on GitHub

Classes

class	truncated documentation
SoftmaxCrossEntropyLoss	Python runtime for function SoftmaxCrossEntropyLoss.

Functions

function	truncated documentation
softmaxcrossentropy	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_softmax_cross_entropy_loss.SoftmaxCrossEntropyLoss(onnx_node, desc=None, **options)

Bases: OpRun

Python runtime for function SoftmaxCrossEntropyLoss.

source on GitHub

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

_run(x, target, weight=None, attributes=None, verbose=0, fLOG=None)

Should be overwritten.

source on GitHub

mlprodict.onnxrt.ops_cpu.op_softmax_cross_entropy_loss.softmaxcrossentropy(x, target, weight=None, reduction='mean', ignore_index=None, get_log_prob=None)

Modified version of softmaxcrossentropy.py to handle other type than float32.

source on GitHub