是否可以使用 2D 数组在图形模式下屏蔽张量像素?

Question

我想用一个由一和零组成的数组来掩盖一个像素数组。 我在图形模式下使用 Keras，因此仅使用可以在张量上执行的操作。

大多数符号运算符似乎在图形模式下工作，并且在keras.backend中有一个numpy操作的子集。但是不支持没有迭代或枚举。

例如，我有：

arr = [[(1,2,3),(2,3,4),(4,5,6)],[(5,6,7),(6,7,8),(7,8,9)]]
mask = [[1, 0, 1],[0, 0, 1]]
# and I want
arr * mask == [[(1,2,3),(0,0,0),(4,5,6)],[(0,0,0),(0,0,0),(7,8,9)]]

实际的数组是图像，比此示例大得多。 有没有合理的方法来掩盖这样的点？

Answer 1

从您提供的代码来看，最简单的方法之一是使用广播，就像注释中提到的@user1318499。由于mask是(2,3)的形状，而arr是(2,3,3)的形状，mask可以展开一个暗淡的来做广播arr。

import tensorflow as tf
arr = tf.reshape(tf.range(2*3*3), [2,3,3])
# array([[[ 0,  1,  2],
#         [ 3,  4,  5],
#         [ 6,  7,  8]],
#        [[ 9, 10, 11],
#         [12, 13, 14],
#         [15, 16, 17]]], dtype=int32)
mask = tf.constant([[1, 0, 1],[0, 0, 1]])
res = arr * tf.expand_dims(mask, axis=-1)
# array([[[ 0,  1,  2],
#         [ 0,  0,  0],
#         [ 6,  7,  8]],
#        [[ 0,  0,  0],
#         [ 0,  0,  0],
#         [15, 16, 17]]], dtype=int32)