tf.keras.layers.Dot

Слой, вычисляющий скалярное произведение между образцами в двух тензорах.

tf.keras.layers.Dot(
    axes, normalize=False, **kwargs
)

Например, если применяется к списку из двух тензоров a и b формы (batch_size, n), выходной тензор будет иметь форму (batch_size, 1), где каждое значение i будет скалярным произведением между a[i] и b[i].

x = np.arange(10).reshape(1, 5, 2)
print(x)
[[[0 1]
  [2 3]
  [4 5]
  [6 7]
  [8 9]]]
y = np.arange(10, 20).reshape(1, 2, 5)
print(y)
[[[10 11 12 13 14]
  [15 16 17 18 19]]]
tf.keras.layers.Dot(axes=(1, 2))([x, y])
<tf.Tensor: shape=(1, 2, 2), dtype=int64, numpy=
array([[[260, 360],
        [320, 445]]])>

x1 = tf.keras.layers.Dense(8)(np.arange(10).reshape(5, 2))
x2 = tf.keras.layers.Dense(8)(np.arange(10, 20).reshape(5, 2))
dotted = tf.keras.layers.Dot(axes=1)([x1, x2])
dotted.shape
TensorShape([5, 1])