TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tfg.math.vector.cross

Computes the cross product between two tensors along an axis.

tfg.math.vector.cross(
    vector1,
    vector2,
    axis=-1,
    name=None
)

Defined in math/vector.py.

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.