tf.linalg.matvec

Умножает матрицу a на вектор b, получая a * b.

tf.linalg.matvec(
    a,
    b,
    transpose_a=False,
    adjoint_a=False,
    a_is_sparse=False,
    b_is_sparse=False,
    name=None
)

Используется в блокнотах

Матрица a после выполнения любых транспонирований должна быть тензором ранга >= 2, с shape(a)[-1] == shape(b)[-1], а shape(a)[:-2] должны иметь возможность вещания с shape(b)[:-1].

Типы a и b должны совпадать. Поддерживаются следующие типы: float16, float32, float64, int32, complex64, complex128.

Матрицу a можно транспонировать или эрмитово сопрягать (сопрягать и транспонировать) непосредственно во время выполнения операции, установив соответствующий флаг в значение True. По умолчанию эти флаги имеют значение False.

Если один или оба входных аргумента содержат много нулей, можно использовать более эффективный алгоритм умножения, установив соответствующий флаг a_is_sparse или b_is_sparse в значение True. По умолчанию эти флаги имеют значение False. Эта оптимизация доступна только для обычных матриц и векторов (тензоров ранга 2 и 1) с типами данных bfloat16 или float32.

Например:

# 2-D tensor `a`
# [[1, 2, 3],
#  [4, 5, 6]]
a = tf.constant([1, 2, 3, 4, 5, 6], shape=[2, 3])

# 1-D tensor `b`
# [7, 9, 11]
b = tf.constant([7, 9, 11], shape=[3])

# `a` * `b`
# [ 58,  139]
c = tf.linalg.matvec(a, b)


# 3-D tensor `a`
# [[[ 1,  2,  3],
#   [ 4,  5,  6]],
#  [[ 7,  8,  9],
#   [10, 11, 12]]]
a = tf.constant(np.arange(1, 13, dtype=np.int32),
                shape=[2, 2, 3])

# 2-D tensor `b`
# [[13, 14, 15],
#  [16, 17, 18]]
b = tf.constant(np.arange(13, 19, dtype=np.int32),
                shape=[2, 3])

# `a` * `b`
# [[ 86, 212],
#  [410, 563]]
c = tf.linalg.matvec(a, b)