索引
索引的基本格式:a[d1][d2][d3]
代码:
取出第一张图片第二个通道中在[20,40]位置的像素点
tensor_h[0][19][39][1]
输出:
如果要提取的索引不连续的话,在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)
输出: