Softmax

Softmax over CHUNK_CNT chunks of CHUNK_SIZE vector. See https://github.com/onnx/onnx/blob/main/docs/Operators.md#Softmax Softmax(input, axis) = Exp(input) / ReduceSum(Exp(input), axis=axis, keepdims=1) More...

Collaboration diagram for Softmax:

Modules
	SoftmaxKernels

Classes
class	SoftmaxGraph< SOFTMAX, INP_H, INP_W, INP_W_PAD >
	Single instance graph. More...

Detailed Description

Template Parameters

SOFTMAX	Softmax Kernel
INP_H	input height
INP_W	input width
INP_W_PAD	input width padded