Google I/O is a wrap! Catch up on TensorFlow sessions View sessions

tfg.math.vector.dot

Computes the dot product between two tensors along an axis.

tfg.math.vector.dot(
    vector1: TensorLike,
    vector2: TensorLike,
    axis: int = -1,
    keepdims: bool = True,
    name: str = 'vector_dot'
) -> TensorLike

Note:

In the following, A1 to An are optional batch dimensions, which should be broadcast compatible.

vector1 Tensor of rank R and shape [A1, ..., Ai, ..., An], where the dimension i = axis represents a vector.
vector2 Tensor of rank R and shape [A1, ..., Ai, ..., An], where the dimension i = axis represents a vector.
axis The dimension along which to compute the dot product.
keepdims If True, retains reduced dimensions with length 1.
name A name for this op which defaults to "vector_dot".

A tensor of shape [A1, ..., Ai = 1, ..., An], where the dimension i = axis represents the result of the dot product.