【解决】torch.jit.frontend.NotSupportedError: slicing multiple dimensions at the same..

Error 信息如下：

raise NotSupportedError(base.range(), “slicing multiple dimensions at the same time isn’t supported yet”)
torch.jit.frontend.NotSupportedError: slicing multiple dimensions at the same time isn’t supported yet
proposals (Tensor): boxes to be encoded
# perform some unpacking to make it JIT-fusion friendly
wx = weights[0]
wy = weights[1]
ww = weights[2]
wh = weights[3]

proposals_x1 = proposals[:, 0].unsqueeze(1)
               ~~~~~~~~~ <--- HERE
proposals_y1 = proposals[:, 1].unsqueeze(1)
proposals_x2 = proposals[:, 2].unsqueeze(1)
proposals_y2 = proposals[:, 3].unsqueeze(1)

reference_boxes_x1 = reference_boxes[:, 0].unsqueeze(1)
reference_boxes_y1 = reference_boxes[:, 1].unsqueeze(1)
reference_boxes_x2 = reference_boxes[:, 2].unsqueeze(1)
reference_boxes_y2 = reference_boxes[:, 3].unsqueeze(1)

解决方式：

1 更新 pytorch 版本，由0.4.1版本到 1.0版本可以解决，或者降到0.2.2版本也行。

2 在 @torch.jit.script 这个注释前加注释 ：# @torch.jit.script，也可以解决！