Spec-Zone.ru › TensorFlow

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

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API