Save the date! Google I/O returns May 18-20 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='vector_cross'
)

Note:

In the following, A1 to An are optional batch dimensions, which should be broadcast compatible.

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".

A tensor of shape [A1, ..., Ai = 3, ..., An], where the dimension i = axis represents the result of the cross product.