Computes the dot product between two tensors along an axis.
tfg.math.vector.dot(
vector1, vector2, axis=-1, keepdims=True, name=None
)
Note:
In the following, A1 to An are optional batch dimensions, which should be
broadcast compatible.
Args |
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".
|
Returns |
A tensor of shape [A1, ..., Ai = 1, ..., An] , where the dimension i = axis
represents the result of the dot product.
|