softmax

[ardeal]

Hi,

Is softmax supported? how to use it in my python code? Is there any document about sofmax in Brevitas?

[MohamedA95]

AKAIK there is no quantized softmax layer but you can use it from torch it's just a mathematical layer.

[ardeal]

@MohamedA95 , Thank you! To use torch.nn.Softmax() in code is not difficult, my question is: the output of the upper layer(qnn.QuantLinear) is quantized data, could it be passed to torch.nn.Softmax() directly?

[MohamedA95]

Hi @ardeal yes it could be passed directly.