tensorflow API学习——tf.strided_slice

最新推荐文章于 2022-03-18 12:28:33 发布

仲夏199603

最新推荐文章于 2022-03-18 12:28:33 发布

阅读量1.5k

点赞数

分类专栏：深度学习

本文链接：https://blog.csdn.net/qq_32458499/article/details/78870401

版权

本文详细介绍了TensorFlow中的tf.strided_slice函数，用于从张量中提取指定步长的切片。该函数允许设置开始、结束索引及步长，支持正向和反向切片，通过示例代码展示了其使用方法。

摘要由CSDN通过智能技术生成

tf.strided_slice(
input_,
begin,
end,
strides=None,
begin_mask=0,
end_mask=0,
ellipsis_mask=0,
new_axis_mask=0,
shrink_axis_mask=0,
var=None,
name=None
)

提取张量的一个跨步切片（广义python数组索引）。

对于第一个命令，此操作end - begin 从由input 指定位置开始的张量中提取一个尺寸的片段begin。切片继续添加 stride到begin索引，直到所有维度不小于end。请注意，步幅的分量可以是负值，这会导致反向切片。

示例代码：

t = tf.constant([[[1, 1, 1], [2, 2, 2]],
                 [[3, 3, 3], [4, 4, 4]],
                 [[5, 5, 5], [6, 6, 6]]])
tf.strided_slice(t, [1, 0, 0], [2, 1,