Computes the cross product between two tensors along an axis.
tfg.math.vector.cross(
vector1, vector2, axis=-1, name=None
)
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.
|
View source on GitHub