module onnxrt.ops_cpu.op_softmax_cross_entropy_loss
#
Short summary#
module mlprodict.onnxrt.ops_cpu.op_softmax_cross_entropy_loss
Runtime operator.
Classes#
class |
truncated documentation |
---|---|
Python runtime for function SoftmaxCrossEntropyLoss. |
Functions#
function |
truncated documentation |
---|---|
Modified version of softmaxcrossentropy.py … |
Properties#
property |
truncated documentation |
---|---|
|
Returns the list of arguments as well as the list of parameters with the default values (close to the signature). … |
|
Returns the list of modified parameters. |
|
Returns the list of optional arguments. |
|
Returns the list of optional arguments. |
|
Returns all parameters in a dictionary. |
Methods#
method |
truncated documentation |
---|---|
Documentation#
Runtime operator.
- class mlprodict.onnxrt.ops_cpu.op_softmax_cross_entropy_loss.SoftmaxCrossEntropyLoss(onnx_node, desc=None, **options)#
Bases:
OpRun
Python runtime for function SoftmaxCrossEntropyLoss.
- __init__(onnx_node, desc=None, **options)#
- _run(x, target, weight=None, attributes=None, verbose=0, fLOG=None)#
Should be overwritten.
- 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.