2 Likes

" The input argument `boxes`

can be either a tensor of length 4, or a two-dimensional tensor of shape (n, 4), where n is the number of bounding boxes."

However, according to the code, the argument boxes can only be a 2-dim tensor of shape(n, 4), since we are using index as [:, 0].