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的使用简介
猜你喜欢
转载自blog.csdn.net/a13602955218/article/details/81254684
今日推荐
周排行