2021年3月5日
指定维度softmax 层tensorRT api实现
摘要： 解决了问题才来记录一下的，现在的心情是好点儿的，但是之前，昨天，真是无厘头，被折腾的一点脾气都没有。 本来就是一个softmax嘛，很简单的嘛，就是按照公式e的指数再相加求和，官方有api实现，比如我找的例子， // Add activation layer using the ReLU algor 阅读全文
posted @ 2021-03-05 14:34 无左无右 阅读(1001) 评论(0) 推荐(0) 编辑