dpnp.tensordot

dpnp.tensordot(x1, x2, axes=2)[source]

Compute tensor dot product along specified axes.

For full documentation refer to numpy.tensordot.

Limitations

Parameters x1 and x2 are supported as dpnp.ndarray. Keyword arguments kwargs are currently unsupported. Parameter axes is supported only with value 1. Otherwise the functions will be executed sequentially on CPU. Input array data types are limited by supported DPNP Data types.

See also

dpnp.dot: Returns the dot product.
dpnp.einsum: Evaluates the Einstein summation convention on the operands.

Examples

>>> import dpnp as np
>>> a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
>>> b = np.array([1, 2, 3])
>>> result = np.tensordot(a, b, 1)
>>> [x for x in result]
[14, 32, 50]