tf.raw_ops.BatchMatMulV2

Перемножает срезы двух тензоров пакетами.

Просмотреть псевдонимы

Псевдонимы для обеспечения совместимости при миграции

Подробнее см. в руководстве по миграции.

tf.compat.v1.raw_ops.BatchMatMulV2

tf.raw_ops.BatchMatMulV2(
    x, y, adj_x=False, adj_y=False, grad_x=False, grad_y=False, name=None
)

Перемножает все срезы Tensor x и y (каждый срез можно рассматривать как элемент пакета) и помещает отдельные результаты в один выходной тензор того же размера пакета. Каждый отдельный срез перед умножением можно при необходимости эрмитово сопрячь (эрмитово сопряжение матрицы означает её транспонирование и комплексное сопряжение), установив флаг adj_x или adj_y в значение True; по умолчанию для них задано значение False.

Входные тензоры x и y имеют размерность 2 или выше и форму [..., r_x, c_x] и [..., r_y, c_y].

Выходной тензор имеет размерность 2 или выше и форму [..., r_o, c_o], где:

r_o = c_x if adj_x else r_x
c_o = r_y if adj_y else c_y

Вычисляется как
output[..., :, :] = matrix(x[..., :, :]) * matrix(y[..., :, :])

Примечание: BatchMatMulV2 поддерживает широковещательное распространение по пакетным размерностям. Подробнее о широковещательном распространении можно прочитать здесь.

Аргументы
`x`	Объект типа `Tensor`. Допустимые типы: `bfloat16`, `half`, `float32`, `float64`, `int16`, `int32`, `int64`, `uint8`, `uint16`, `uint32`, `uint64`, `complex64`, `complex128`. Размерность 2 или выше, форма `[..., r_x, c_x]`.
`y`	Объект типа `Tensor`. Должен иметь тот же тип, что и `x`. Размерность 2 или выше, форма `[..., r_y, c_y]`.
`adj_x`	Необязательный объект типа `bool`. Значение по умолчанию: `False`. Если `True`, эрмитово сопрягает срезы `x`. Значение по умолчанию: `False`.
`adj_y`	Необязательный объект типа `bool`. Значение по умолчанию: `False`. Если `True`, эрмитово сопрягает срезы `y`. Значение по умолчанию: `False`.
`grad_x`	Необязательный объект типа `bool`. Значение по умолчанию: `False`.
`grad_y`	Необязательный объект типа `bool`. Значение по умолчанию: `False`.
`name`	Имя операции (необязательно).

Возвращает
Объект типа `Tensor`. Имеет тот же тип, что и `x`.

© 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/raw_ops/BatchMatMulV2