stn算子_深度学习常用算子(二)

1、Tensor维度变换

1)Flatten

作用:将输入tensor中从start_axis维度到end_axis维度合并为1维

2)Reshape

作用:将输入Tensor描述转换为新的shape

3)FreespaceExtract

作用:将h维变成1,其他维度不变,从而完成对h的采样,采样值所在位置由输入的index参数决定

4)Pack

作用:Pack算子为TensorFlow原生算子,最新的版本已经改名为:Stack。该算子以指定的轴axis,将一个维度为R的张量数组转变成一个维度为R+1的张量。

5)Pad

作用:进行数据扩充

6)Permute

作用:调整Tensor的输入维度顺序

7)ShuffleChannel

作用:调整C维的排序

使用网络:ShuffleNet

8)Title

作用:将输入数据在每个维度上复制指定次数来生成输出数据

2、Tensor单个元素运算

1)Rsqrt

公式:y = 1 / sqrt{x}

2)Scale

公式:y(x)=scale*x+bias

3)AbsVal

公式:y(x)=|x|

4)Power

公式:f(x)= (scale * x + shift) ^ power

3、分类

Softmax

公式:

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值