TensorFlow 切片操作

1、切片操作

import tensorflow as tf
import numpy as np
print(tf.__version__)
print(np.__version__)

# 切片 start:end  [A:B)
a = tf.range(10)
print(a)

print(a[-1:])
print(a[-2:])
print(a[:2])
print(a[:-1])

2、部分维度切片

# 取部分维度数据
a = tf.random.normal([4, 28, 28, 3])
print(a.shape)

print(a[0].shape)
print(a[0,1,:,:].shape)
print(a[:,:,:,0].shape)
print(a[:,:,:,2].shape)
print(a[:,0,:,:].shape)
print(a[0:2,:,:,:].shape)

3、隔行(step)切片采样

# 步长 隔行采样[A:B:step) 
print(a[:,0:28:2,0:28:2,:].shape)

print(a[:,:14,:14,:].shape)
print(a[:,14:,14:,:].shape)
print(a[:,::2,::2,:].shape)
print(a[:,2:26:2,2:26:2,:].shape)

4、逆序切片

# 逆序切片 [B:A:-1)
a = tf.range(4)
print(a)

print(a[::-1])
print(a[::-2])
print(a[2::-2])

5、截取一列

sub = tf.random.uniform((3, 3)) # 随机创建 3*3的数组
print(sub.numpy()) # 以numpy格式显示
print(sub[..., 1].numpy(), sub[..., 1].shape) # 截取第二列

索引

a = tf.random.normal([4, 35, 8])
print(a.shape)

# 数据 0维度数据  索引2，3 组合的数据
b1 = tf.gather(a, axis=0, indices=[2, 3])
print(b1.shape)

# 重新排列 0维数据  [0, 1, 2, 3] ==> [1, 0, 3, 2]
b = tf.gather(a, axis=0, indices=[1, 0, 3, 2])
print(b.shape)

# 抽取 1维 7，9，15 索引数据
c = tf.gather(a, axis=1, indices=[7, 9, 15])
print(c.shape)

# 抽取 2维 3,4,6 索引数据
d = tf.gather(a, axis=2, indices=[3, 4, 6])
print(d.shape)

# tf.gather_nd
# 抽取 给定索引值数据 组合
b = tf.gather_nd(a, [0, 1])
print(b.shape)

c = tf.gather_nd(a, [0, 1, 2])
print(c.shape)

d = tf.gather_nd(a, [[0, 1, 2]])
print(d.shape)

# 抽取 不同索引数据 组合
b = tf.gather_nd(a, [[0, 1], [1, 1]])
print(b.shape)

c = tf.gather_nd(a, [[0, 0, 0], [1, 1, 1], [2, 2, 2]])
print(c.shape)

d = tf.gather_nd(a, [[[0, 0, 0], [1, 1, 1], [2, 2, 2], [3, 3, 3]]])
print(d.shape)

# tf.boolean_mask
a = tf.random.normal([4, 28, 28, 3])
print(a.shape)

# 抽取 True所在索引的数据
b = tf.boolean_mask(a, mask=[True,True,False,False])
print(b.shape)

c = tf.boolean_mask(a, mask=[True,True,False], axis=3)
print(c.shape)

a = tf.ones([2, 3, 4])

d = tf.boolean_mask(a, mask=[[True, False, False], [False, True, True]])
print(d.shape)