tf.strided_slice函数(类似K.shape(feats)[1:3])

在深度学习框架TensorFlow中,`tf.strided_slice`用于从张量中提取特定步长的切片。文章通过举例说明了如何使用该函数,特别是在Keras_YOLO模型中的应用,以及在CIFAR10数据集中的实例。通过`tf.strided_slice(input, [0,0,0], [2,2,2], [1,2,1])`的调用,解释了start、end和stride参数的作用,并展示了如何从三维张量中提取特定部分,最后给出了输出结果`[[[1,1],[3,3]]]`。" 126611476,15081818,深入理解JavaScript正则表达式,"['正则表达式', 'javascript', '前端开发']
摘要由CSDN通过智能技术生成

在keras_yolo中model函数下grid_shape = K.shape(feats)[1:3]
grid_shape:
<tf.Tensor ‘strided_slice:0’ shape=(0,) dtype=int32>
cifar10的例子中也有。

Defined in tensorflow/python/ops/array_ops.py.

input
来把输入变个型,可以看成3维的tensor,从外向里为1,2,3维
(维数的判断顺序为z轴–y轴–x轴)

[[

      [1,1,1]

      [2,2,2]

 ]

 [

     [3,3,3]

     [4,4,4]

 ]

 [

     [5,5,5]

     [6,6,6]

 ]]

以tf.strided_slice(input, [0,0,0], [2,2,2], [1,2,1])调用为例,start = [0,0,0] , end = [2,2,2], stride = [1,2,1],求一个[start, end)的一个片段,注意end为开区间

第1维 start = 0 , end = 2, stride = 1, 所以取 0 , 1行,此时的输出

output1=

[ [

      [1,1,1]

      [2,2,2]

 ]

 [

     [3,3,3]

     [4,4,4]

 ]]

第2维时, start = 0 , end = 2 , stride = 2, 所以只能取0行,此时的输出

output2=

[ [

      [1,1,1]

 ]

 [

     [3,3,3]

 ]]

第3维的时候,start = 0, end = 2, stride = 1, 可以取0,1行,此时得到的就是最后的输出

 [[

      [1,1]

 ]

 [

     [3,3]

 ]]

整理之后最终的输出为:

[[[1,1],[3,3]]]

更多例子:
非严谨算output_shape:|end - start|

t = tf.constant([[[1, 1, 1], <
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值