torch_npu.npu_anchor_response_flags

torch_npu.npu_anchor_response_flags(self, featmap_size, stride, num_base_anchors) -> Tensor

在单个特征图中生成锚点的责任标志。

• 参数解释：
  • self (Tensor) - 真值框，shape为[batch, 4]的2D张量。
  • featmap_size (ListInt of length 2) - 特征图大小。
  • strides (ListInt of length 2) - 当前水平的步长。
  • num_base_anchors (Int) - base anchors的数量。
• 约束条件：

  无

• 示例：

  >>> x = torch.rand(100, 4).npu()
  >>> y = torch_npu.npu_anchor_response_flags(x, [60, 60], [2, 2], 9)
  >>> y.shape
  torch.Size([32400])