如何索引pytorch中的张量？

Question

我在pytorch方面遇到了一些问题。 当计算mu和sig时，出现以下消息： ValueError：切片结果是空张量

这是我的密码：

x            = mu + sig*torch.randn(N,1)
S            = -x**2 + 100
sorted_v , I = torch.sort(S,0)
mu           = alpha*torch.mean(x[I[(1-rho)*N:N]]) + (1-alpha)*mu
sig          = alpha*torch.std(x[I[(1-rho)*N:N]])  + (1-alpha)*sig

在MATLAB我会这样做：

mu    = alpha*mean(x(I((1-rho)*N:N))) + (1-alpha)*mu;
sig   = alpha*std(x(I((1-rho)*N:N)))  + (1-alpha)*sig;

怎么了？为什么？ 太感谢你了。你知道吗

Answer 1

因为你的代码是不可运行的，我不能给你解决这个错误的方法。但我能告诉你为什么你可能会出错。你知道吗

在以下两行代码中：

mu  = alpha * torch.mean(x[I[(1-rho)*N:N]]) + (1-alpha) * mu
sig = alpha * torch.std(x[I[(1-rho)*N:N]]) + (1-alpha) * sig

I[(1-rho)*N:N]或x[I[(1-rho)*N:N]]返回一个空张量。理解错误原因的一种简单方法是打印x和I的形状。并且还打印rho和N的值。你知道吗