使用索引对tensor进行切片修改

最新推荐文章于 2024-10-20 11:02:14 发布

旭旭想开点

最新推荐文章于 2024-10-20 11:02:14 发布

阅读量223

点赞数

文章标签： python Powered by 金山文档

本文链接：https://blog.csdn.net/m0_59795442/article/details/129413982

版权

场景：有一个tensor(4,2):tensor([[4,15,15,19],[19,13,23,15]])作为索引代表的是两个矩形框的left,top,right,bottom，我想在一个（34，25）的0tensor中将上述矩形框中的值置为1

方案：

import torch

# 创建形状为 (34, 25) 的张量，所有元素均为 0
result = torch.zeros((34, 25))

# 遍历每个矩形框
for i in range(bbox.shape[0]):
    # 获取当前矩形框的左上和右下坐标
    left, top, right, bottom = bbox[i]

    # 将矩形框中的元素全部置为 1
    result[top:bottom+1, left:right+1] = 1

在上面的代码中，我们首先创建了一个形状为 (34, 25) 的张量 result，所有元素都设置为 0。然后，我们遍历了每个矩形框，获取了当前矩形框的左上和右下坐标，并将矩形框中的所有元素都设置为 1。

注意，在使用索引对张量进行修改时，我们需要使用切片（slice）来指定需要修改的元素范围。在本例中，我们使用了 result[top:bottom+1, left:right+1] 来获取需要修改的元素。这里使用了 bottom+1 和 right+1 是因为 Python 中的切片是左闭右开区间，不包括右边界。