tf.stride_slice(data, begin, end)
tf.slice(data, begin, end)
两者都是用于切割矩阵,不同点在于slice的end索引是闭区间,stride_slice的end索引是开区间,所以一个截掉最后一列的小技巧是用stride_slice(data, [0, 0], [rows, -1]),但是如果是用slice(data, [0, 0], [rows, -1])则仍是原矩阵。
注意begin和end不是一个常数,而是一个数组,
例如对于矩阵[1,2,3,4,5],截取前三个,应该用tf.slice(data,[0],[2])
对于
[[1,2,3],
[4,5,6],
[7,8,9]]
截取1,2,4,5,应该用tf.slice(data,[0,0],[1,1])
tf.stride_slice和tf.slice的使用简介
最新推荐文章于 2022-03-18 12:28:33 发布