Rank-agnostic softmax over the last axis, with AttnLRP's Eq. 13 DTD relevance rule. More...

#include <optional>
#include "pulsatrix/module.hpp"

Include dependency graph for softmax_module.hpp:

This graph shows which files directly or indirectly include this file:

Classes
class	pulsatrix::SoftmaxModule
	Softmax over the tensor's last dimension, applied independently to every "row" (every fixed combination of all leading dimensions). More...

Detailed Description

Rank-agnostic softmax over the last axis, with AttnLRP's Eq. 13 DTD relevance rule.

Classes