张量流中具有标量的批量乘法/除法

Question

我正在努力寻找一种简单的方法将一批张量与一批标量相乘。

我有一个张量为N，4、4的张量。我想要的是将批次中的张量除以位置3、3上的值。

例如，假设我有：

A = [[[1, 1, 1, 0],
      [1, 1, 1, 0],
      [1, 1, 1, 0],
      [0, 0, 0, a]], 
     [[1, 1, 1, 0],
      [1, 1, 1, 0],
      [1, 1, 1, 0],
      [0, 0, 0, b]]

我想要获得以下内容：

 B = [[[1/a, 1/a, 1/a, 0],
      [1/a, 1/a, 1/a, 0],
      [1/a, 1/a, 1/a, 0],
      [0, 0, 0, 1]], 
     [[1/b, 1/b, 1/b, 0],
      [1/b, 1/b, 1/b, 0],
      [1/b, 1/b, 1/b, 0],
      [0, 0, 0, 1]]

Answer 1

您应该这样做：

B = A / A[:, 3:, 3:]