tf.keras.layers.Dot
| Просмотреть исходный код на GitHub |
Слой, вычисляющий скалярное произведение между образцами в двух тензорах.
tf.keras.layers.Dot(
axes, normalize=False, **kwargs
)
Например, если применяется к списку из двух тензоров a и b формы (batch_size, n), выходной тензор будет иметь форму (batch_size, 1), где каждый элемент i будет скалярным произведением между a[i] и b[i].
| Аргументы | |
|---|---|
axes | Целое число или кортеж целых чисел, ось или оси, по которым вычисляется скалярное произведение. |
normalize | Нужно ли L2-нормализовать образцы по оси скалярного произведения перед вычислением скалярного произведения. Если установлено в True, то выходное значение скалярного произведения будет косинусной близостью между двумя образцами. |
**kwargs | Стандартные ключевые аргументы слоя. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/keras/layers/Dot
