numpy/tensorflow筛选/过滤得到array/tensor的shape注意点

先阐述结论:多维数组通([None, ...(M-1次), N])过其他数组([None, ...(M次)])过滤时,得到新的array是None * N大小 

示例如下:

n1 = np.random.randint(1,5, size=[2,3,4,5])

n2 = np.random.randint(1,5, size=[2,3,4])

 n1[n2 > 2].shape

其他示例:

np.arange(9).reshape(3, 3)

a[a[:,0] > 3]

a[np.where(a[:,0] > 3)]

-- over --

猜你喜欢

转载自blog.csdn.net/qm5132/article/details/83345183