在python中获取作为整数的Pytorch-tensor值

Question

我有我的torch张量的输出，它看起来像下面

(物体检测中边界框的坐标(

[tensor(299., device='cuda:0'), tensor(272., device='cuda:0'), tensor(327., device='cuda:0'), tensor(350., device='cuda:0')]

我想提取每个张量值作为minx，miny，maxx，maxy
形式的int，这样我就可以将其传递给以下形式的形状函数

from shapely.geometry import box
minx,miny,maxx,maxy=1,2,3,4
b = box(minx,miny,maxx,maxy)

最好的方法是什么？通过避免、Cuda是否启用或其他异常？

Answer 1

minx, miny, maxx, maxy = [int(t.item()) for t in tensors]

其中CCD_ 1是张量的列表。