代码之家 › 专栏 › 技术社区 › Rotacional

获取矩阵中的方向元素

torch signal-processing matrix python

Rotacional · 技术社区 · 11 月前

假设我的矩阵中有一个感兴趣的点,即NxN。该点位于位置ij。那么,给定索引ij,有没有一种简单的方法可以让线元素通过ij到达原点(位于矩阵的中间)?

我正在使用torch,我认为使用torch.diag将是第一个开始,但实际上这个函数不会在矩阵的中间通过。

def directionalK(kx,ky, indices):
    '''Function that provides the K values at a given direction dictated by the indices'''
    kx_grid,ky_grid = torch.meshgrid(kx,kx, indexing='ij')
    k_grid = torch.sqrt(kx_grid**2 + ky_grid**2) 
    k_grid[...,:len(k_grid)//2] *=-1 
    y,x = indices
    diag = x - len(k_grid)//2

1 回复 | 直到 11 月前

Warren Weckesser 11 月前

我想你可以用 skimage.draw.line 方法,您输入开始(i,j)和结束(0,0)坐标,它计算属于可用于索引的线的索引。

工作示例:

from skimage.draw import line
import numpy as np
import matplotlib.pyplot as plt
arr = np.zeros((100, 100))
i, j = 10, 80
origin = 50, 50
rr, cc = line(*origin, i, j)
arr[rr, cc] = 1
plt.imshow(arr, cmap='gray', interpolation='nearest')

或者,如果您希望该行继续:

arr = np.zeros((100, 100))
rr, cc = line(i, j, 2*origin[0]-i, 2*origin[1]-j)
arr[rr, cc] = 2
plt.imshow(arr, cmap='gray', interpolation='nearest')

推荐文章

Qinqing Liu · 如何构造一个三维张量,其中每个二维子张量是PyTorch中的对角矩阵?

7 年前

Elias E. · pytorch神经网络中的交叉熵损失。交叉熵()

7 年前

Wasi Ahmad · 参数维数对聚集函数的影响

7 年前

Salih Karagoz · 如何为图像命名。显示窗口?

7 年前

Beanocean · hdf5缺少依赖项:图腾错误:克隆git存储库失败;git clome超时

7 年前

floyd · 为什么我不能使用转置函数没有:?

7 年前

ytrewq · 从火炬中删除项目。张量

7 年前

Minkyu Choi · 当有两个gpu时,如何设置火炬仅使用一个gpu?

9 年前

Amir · 如何计算火炬中相对于任意层/重量的损失梯度?

9 年前

Alejandro Simkievich · 如何从python获取lua/翻译中的类类型

9 年前