numpy.dot#
- numpy.dot(a, b, выход=None)#
Скалярное произведение двух массивов. Конкретно,
Если оба a и b являются 1-D массивами, это скалярное произведение векторов (без комплексного сопряжения).
Если оба a и b являются 2-D массивами, это матричное умножение, но с использованием
matmulилиa @ bпредпочтительнее.Если любой из a или b является 0-D (скаляром), это эквивалентно
multiplyи использованиеnumpy.multiply(a, b)илиa * bявляется предпочтительным.Если a является N-мерным массивом и b является одномерным массивом, это сумма произведений по последней оси a и b.
Если a является N-мерным массивом и b является M-D массивом (где
M>=2), это суммарное произведение по последней оси a и предпоследней осью b:dot(a, b)[i,j,k,m] = sum(a[i,j,:] * b[k,:,m])
Он использует оптимизированную библиотеку BLAS, когда это возможно (см.
numpy.linalg).- Параметры:
- aarray_like
Первый аргумент.
- barray_like
Второй аргумент.
- выходndarray, необязательно
Выходной аргумент. Он должен иметь точный вид, который был бы возвращен, если бы не использовался. В частности, он должен иметь правильный тип, быть C-непрерывным, и его dtype должен быть dtype, который был бы возвращен для dot(a,b). Это функция производительности. Поэтому, если эти условия не выполнены, возникает исключение, вместо попытки быть гибким.
- Возвращает:
- выводndarray
Возвращает скалярное произведение a и b. Если a и b оба скаляры или оба 1-D массива, то возвращается скаляр; в противном случае возвращается массив. Если выход если задано, то возвращается.
- Вызывает:
- ValueError
Если последнее измерение a не имеет того же размера, что и предпоследнее измерение b.
Смотрите также
Примеры
>>> import numpy as np >>> np.dot(3, 4) 12
Ни один из аргументов не является комплексно-сопряжённым:
>>> np.dot([2j, 3j], [2j, 3j]) (-13+0j)
Для 2-D массивов это матричное произведение:
>>> a = [[1, 0], [0, 1]] >>> b = [[4, 1], [2, 2]] >>> np.dot(a, b) array([[4, 1], [2, 2]])
>>> a = np.arange(3*4*5*6).reshape((3,4,5,6)) >>> b = np.arange(3*4*5*6)[::-1].reshape((5,4,6,3)) >>> np.dot(a, b)[2,3,2,1,2,2] 499128 >>> sum(a[2,3,2,:] * b[1,2,:,2]) 499128