tf.keras.layers.Dot
Вычисляет поэлементное скалярное произведение двух тензоров.
Наследуется от: Layer, Operation
tf.keras.layers.Dot(
axes, normalize=False, **kwargs
)
Используется в блокнотах
| Используется в руководствах |
|---|
Принимает список входных данных размером 2 и оси, соответствующие каждому входу, вместе с скалярным произведением, которое должно быть выполнено.
Допустим, x и y — это два входных тензора с формами (2, 3, 5) и (2, 10, 3). Размерность пакета должна быть одинаковой для обоих входов, и axes должна соответствовать размерностям, которые имеют одинаковый размер в соответствующих входах. Например, с axes=(1, 2), скалярное произведение x и y приведет к тензору с формой (2, 5, 10)
Пример:
x = np.arange(10).reshape(1, 5, 2) y = np.arange(10, 20).reshape(1, 2, 5) keras.layers.Dot(axes=(1, 2))([x, y])
Использование в модели Keras:
x1 = keras.layers.Dense(8)(np.arange(10).reshape(5, 2)) x2 = keras.layers.Dense(8)(np.arange(10, 20).reshape(5, 2)) y = keras.layers.Dot(axes=1)([x1, x2])
| Аргументы | |
|---|---|
axes | Целое число или кортеж целых чисел, ось или оси, по которым берется скалярное произведение. Если кортеж, должен содержать два целых числа, соответствующие желаемой оси из первого входа и желаемой оси из второго входа соответственно. Обратите внимание, что размер двух выбранных осей должен совпадать. |
normalize | Нужно ли L2-нормализовать образцы вдоль оси скалярного произведения перед взятием скалярного произведения. Если установлено в True, то выходное значение скалярного произведения является косинусной близостью между двумя образцами. |
**kwargs | Стандартные ключевые аргументы слоя. |
| Возвращаемое значение | |
|---|---|
| Тензор, скалярное произведение образцов из входов. |
| Атрибуты | |
|---|---|
input | Получает тензор(ы) ввода символической операции. Возвращает только тензор(ы), соответствующие первому вызову операции. |
output | Получает тензор(ы) вывода слоя. Возвращает только тензор(ы), соответствующие первому вызову операции. |
Методы
from_config
@classmethod
from_config(
config
)
Создает слой по его конфигурации.
Этот метод является обратным к get_config, способный создать тот же слой из словаря конфигурации. Он не обрабатывает соединение слоев (обрабатывается сетью), а также веса (обрабатываются set_weights).
| Аргументы | |
|---|---|
config | Словарь Python, обычно результат get_config. |
| Возвращаемое значение | |
|---|---|
| Экземпляр слоя. |
symbolic_call
symbolic_call(
*args, **kwargs
)
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/layers/Dot