切片
切片的方式主要有:
[start: end]:从tensor的开始位置到结束位置的数据切片;
[start :end :step]或者[::step]:从tensor的开始位置到结束位置每隔step的数据切片;
[::-1]:负数表示倒序切片;
‘...’:任意长。
代码:
创建一个4维tensor。tensor包含4张图片,每张图片的大小为1001003
tensor_h = tf.random.normal([4,100,100,3])
tensor_h
输出:
如果要提取的索引不连续的话,在TensorFlow中,常见的用法为tf.gather和tf.gather_nd。
在某一维度进行索引。
tf.gather(params, indices,axis=None):
params:输入张量;
indices:取出数据的索引;
axis:所取数据所在维度。
代码:
取出tensor_h([4,100,100,3])中,第1,2,4张图像。
indices = [0,1,3]
tf.gather(tensor_h,axis=0,indices=indices,batch_dims=1)
输出: