在TensorFlow 1.12中,可以使用tf.slice函数从2D张量中获取张量切片。tf.slice函数的语法如下:
tf.slice(input_, begin, size, name=None)
其中,input_表示输入的2D张量,begin表示切片的起始位置,size表示切片的大小。这些参数都是张量,可以是常量或变量。
下面是一个示例代码,展示如何在TensorFlow 1.12中从2D张量中获取张量切片:
import tensorflow as tf
# 创建一个2D张量
input_tensor = tf.constant([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 获取张量切片
slice_tensor = tf.slice(input_tensor, [1, 1], [2, 2])
# 创建会话并运行
with tf.Session() as sess:
result = sess.run(slice_tensor)
print(result)
运行以上代码,将输出以下结果:
[[5 6]
[8 9]]
在这个示例中,我们创建了一个2D张量input_tensor
,然后使用tf.slice
函数从第二行第二列开始,获取大小为2x2的张量切片slice_tensor
。最后,通过运行会话,我们获得了切片的结果。
推荐的腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云