tensorflow中的切片操作

tf.slice(input_, begin, size)

import tensorflow as tf

# 创建一个形状为(3, 3)的二维张量
x = tf.constant([[1, 2, 3],
                 [4, 5, 6],
                 [7, 8, 9]])

# 提取第一行
row_1 = tf.slice(x, [0, 0], [1, 3])
print(row_1)  # 输出: [[1 2 3]]

# 提取第二行
row_2 = tf.slice(x, [1, 0], [1, 3])
print(row_2)  # 输出: [[4 5 6]]

import tensorflow as tf

# 创建一个形状为(3, 3)的二维张量
x = tf.constant([[1, 2, 3],
                 [4, 5, 6],
                 [7, 8, 9]])

# 提取第一列
col_1 = tf.slice(x, [0, 0], [3, 1])
print(col_1)  # 输出: [[1], [4], [7]]

# 提取第二列
col_2 = tf.slice(x, [0, 1], [3, 1])
print(col_2)  # 输出: [[2], [5], [8]]

import tensorflow as tf

# 创建一个形状为(3, 3)的二维张量
x = tf.constant([[1, 2, 3],
                 [4, 5, 6],
                 [7, 8, 9]])

# 提取左上角的2x2子矩阵
sub_matrix = tf.slice(x, [0, 0], [2, 2])
print(sub_matrix)  # 输出: [[1 2], [4 5]]