我在python中有一个for循环,我想将其展开到GPU上。我想一定会有一个简单的解决方案,但我还没有找到。我们的函数循环遍历一个numpy数组中的元素,并执行一些数学运算,将结果存储在另一个numpy数组中。每次迭代都会向这个结果数组中添加一些。我们的代码可能会有如下所示的大规模简化:out = np.array([0, 0])
for x in xra
此建议我们可以使用Python列表向其添加元素,并使用长度未知的np.array将其转换为numpy数组,如下所示:for x in y:a = np.array(a)
假设x是一个形状为(h,w)的numpy.ndarray,但h和w是未知的,我们如何在循环中向ndarray a的第二轴添加元素,使a变为(h,w,d),其中d是迭代的次数?我知道我们可以使用dstack将元素加在一起