tfg.math.vector.dot

Stay organized with collections Save and categorize content based on your preferences.

Computes the dot product between two tensors along an axis.

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.