1. 程式人生 > >tf.stride_slice和tf.slice的使用簡介

tf.stride_slice和tf.slice的使用簡介

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])