How to use the softmax(dim=1)?

[smilecode-github]

Hello.

I am making a segmentation model using softmax(dim=1).

I compiled my model using edgeai-tidl-tools on PC.

But, I found softmax(dim=1) layer is not supported.

Because softmax is only supported in [1, 1, 1, N], so it cannot be used for dim=1.

How to use the softmax(dim=1)?

Also, How to check softmax(dim=1) in compile using edgeai-tidl-tools on PC?

[suyali]

@smilecode-github Have you solved this problem?

[vtrip97]

@smilecode-github : You could offload the particular layer via deny-list . Alternatively, you could keep this on the DSP by representing it as a combination of transpose and reshape layers (and still perform softmax on the outermost dimension)