毕火炬张量**2计算

Question

for i in range(180000):
    p[indices[0][i], :, indices[1][i], indices[2][i]] = \
        p[indices[0][i], :, indices[1][i], indices[2][i]] ** 2

它需要假设60+秒数，而p = p ** 2现在就可以达到这个结果。我只想改变P的一部分。那么，我该怎么做才能达到这样的效率呢？编码由Py手电筒.

Answer 1

也许我已经知道原因了。我应该将索引转换为cuda格式。