Why need implement softmax plugin?

Open ujsyehao opened this issue 7 years ago • 1 comments

In tensorRT 4.0 doc: It supports softmax on C, H, W dimension

Dec 05 '18 17:12 ujsyehao

| | 卢晨光 | | 邮箱：[email protected] |

签名由网易邮箱大师定制

On 12/06/2018 01:41, yehao wrote:

In tensorRT 4.0 doc: It supports softmax on C, H, W dimension

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub, or mute the thread.

It does not support softmax(axis=2)

Dec 05 '18 22:12 Ghustwb