Python tf.Tensor.__getitem__用法及代碼示例

用法

__getitem__(
    slice_spec, var=None
)

Tensor.getitem 的重載。

此操作從張量中提取指定區域。該表示法類似於 NumPy，但目前僅支持基本索引。這意味著當前不允許使用非標量張量作為輸入。

一些有用的例子：

# Strip leading and trailing 2 elements
foo = tf.constant([1,2,3,4,5,6])
print(foo[2:-2].eval())  # => [3,4]

# Skip every other row and reverse the order of the columns
foo = tf.constant([[1,2,3], [4,5,6], [7,8,9]])
print(foo[::2,::-1].eval())  # => [[3,2,1], [9,8,7]]

# Use scalar tensors as indices on both dimensions
print(foo[tf.constant(0), tf.constant(2)].eval())  # => 3

# Insert another dimension
foo = tf.constant([[1,2,3], [4,5,6], [7,8,9]])
print(foo[tf.newaxis,:,:].eval()) # => [[[1,2,3], [4,5,6], [7,8,9]]]
print(foo[:, tf.newaxis,:].eval()) # => [[[1,2,3]], [[4,5,6]], [[7,8,9]]]
print(foo[:,:, tf.newaxis].eval()) # => [[[1],[2],[3]], [[4],[5],[6]],
[[7],[8],[9]]]

# Ellipses (3 equivalent operations)
foo = tf.constant([[1,2,3], [4,5,6], [7,8,9]])
print(foo[tf.newaxis,:,:].eval())  # => [[[1,2,3], [4,5,6], [7,8,9]]]
print(foo[tf.newaxis, ...].eval())  # => [[[1,2,3], [4,5,6], [7,8,9]]]
print(foo[tf.newaxis].eval())  # => [[[1,2,3], [4,5,6], [7,8,9]]]

# Masks
foo = tf.constant([[1,2,3], [4,5,6], [7,8,9]])
print(foo[foo > 2].eval())  # => [3, 4, 5, 6, 7, 8, 9]

注意:

• tf.newaxis 是 None，就像在 NumPy 中一樣。
• 隱式省略號放置在slice_spec 的末尾
• 目前不支持 NumPy 高級索引。

API 中的用途：

此方法在 TensorFlow 的 API 中公開，因此庫開發人員可以為 Tensor.getitem 注冊調度，以允許它處理自定義複合張量和其他自定義對象。

API 符號不打算由用戶直接調用，並且確實出現在 TensorFlow 生成的文檔中。