Make the SPNE softmax layer support the 4D model with axis=3

实现SNPE softmax layer的4D tensor 第4轴的排序

SNPE的softmax只支持1D和2D的输入,如何实现4D数据的排序呢? 并且SNPE的排序的轴是固定的axis=-1,如何对4D数据的第3个轴排序呢?

def softmax(data, axis=3):
    dt_shape = data.get_shape().as_list()
    dt_axis1 = tf.split

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览