Как замаскировать массив с другим массивом?

Question

x = np.array([
  [[1,2,3], [4,5,6], [1,2,1]],
  [[1,2,3], [4,5,6], [3, 5, 8]]
])

z = np.array([
  [1, 0, 1],
  [0,1, 0]
])

#output
[
  [[1,2,3], [0,0,0], [1,2,1]],
  [[0,0,0], [4,5,6], [0,0,0]]
]

Как сгенерировать этот вывод из x и z, используя numpy встроенные функции (без for loop)?