tfm.vision.iou_similarity.area

Computes area of boxes.

B: batch_size N: number of boxes

box a float Tensor with [N, 4], or [B, N, 4].

a float Tensor with [N], or [B, N]