Numpy 2-D布尔掩码

Question

我正在尝试为2-D numpy数组形状（100,785）创建一个布尔掩码，但我的面具出现为（100,1）。

ywx = np.multiply(y_t, x_t @ w)
mask = ywx < 1 

其中mask是我的（100,1）布尔掩码。 我目前正在做以下扩展它：

expanded_mask = np.array([np.repeat(boolean[0], 785) for boolean in mask])

有更好的方法吗？