Computes the cross product between two tensors along an axis.
tfg.math.vector.cross(
vector1: TensorLike,
vector2: TensorLike,
axis: int = -1,
name: str = 'vector_cross'
) -> TensorLike
Note |
In the following, A1 to An are optional batch dimensions, which should be
broadcast compatible.
|
Args |
vector1
|
A tensor of shape [A1, ..., Ai = 3, ..., An] , where the dimension
i = axis represents a 3d vector.
|
vector2
|
A tensor of shape [A1, ..., Ai = 3, ..., An] , where the dimension
i = axis represents a 3d vector.
|
axis
|
The dimension along which to compute the cross product.
|
name
|
A name for this op which defaults to "vector_cross".
|
Returns |
A tensor of shape [A1, ..., Ai = 3, ..., An] , where the dimension i = axis
represents the result of the cross product.
|