- 博客(2)
- 收藏
- 关注
原创 torch.nn.functional.grid_sample()详解
简单来说就是,提供一个input的Tensor以及一个对应的flow-field网格(比如光流,等),然后根据grid中每个位置提供的坐标信息(这里指input中pixel的坐标),将input中对应位置的像素值填充到grid指定的位置,得到最终的输出。是 PyTorch 中的一个函数,用于对输入图像或张量进行采样和插值。该函数通过指定的网格坐标 (grid) 来图像中的像素,实现图像的变换(如仿射变换、透视变换等)。
2025-03-26 14:30:44
967
原创 torch.gather()函数详解
那么j如何获得呢?从index of index中拿到,index每一个元素的索引为(0,0) (0,1) (0,2),取j,则为0,1,2,那么输出则为。,index每一个元素的索引为(0,0) (0,1) (0,2),i均为1,那么。即input是一个矩阵,根据dim的值,将index的值替换到不同的维度的。例子:首先我们生成3×3的矩阵,明确行索引的概念,第0行指的是。,当dim为0时,index替代i的值,成为第0维度的索引。维度为1,则替换列索引的值,那么输出为。,替换了i的索引,为。
2025-03-26 13:44:26
224
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人