Ghustwb
commented
5 years ago | 卢晨光 | |
|---|---|
邮箱：lcgcf2013@163.com |
签名由 网易邮箱大师 定制
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)
In tensorRT 4.0 doc: It supports softmax on C, H, W dimension