二维数组矩阵与一维数组转换后，index索引变化问题

最新推荐文章于 2022-04-03 16:29:57 发布

zhuhaow

最新推荐文章于 2022-04-03 16:29:57 发布

阅读量429

点赞数

文章标签：算法线性代数 python

本文链接：https://blog.csdn.net/qq_40600539/article/details/119901041

版权

题目：给定H * W的一维数组，指定index，求reshape成二维矩阵后，对应的二维矩阵index（x, y）。

>>> import torch
>>> x = torch.tensor([1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16])
>>> x
tensor([ 1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15, 16])
>>> x.reshape([8,2])
tensor([[ 1,  2],
        [ 3,  4],
        [ 5,  6],
        [ 7,  8],
        [ 9, 10],
        [11, 12],
        [13, 14],
        [15, 16]])

求的方法。

h = index // W

w = index % W

例如：index = 11时，h = 11 // 8 =1 , w = 11%8=3，故索引为（1，3）