是否存在numpy.view_as_windows的张量流版本？

Question

我正在使用python3和tensorflow来使用numpy view_as_windows生成图像补丁，但由于numpy无法在GPU上运行，有没有办法用tensorflow来做？

ex：view_as_windows(array2d, window_shape, stride)

由于

Answer 1

注意：这个答案不回答了OP的确切问题，但是在评论中阐明了OP的实际需求（即快速生成图像补丁） ）。我只是认为这比在格式错误的评论中更适合。

如果您只需要生成图像补丁，Tensorflow（通常是GPU加速）就不是正确的工具，因为实际的计算是微不足道的（提取一个子区域）图像）和瓶颈将是GPU和CPU之间的内存传输。

然后，我的建议是编写使用view_as_windows的纯CPU代码，并通过multiprocessing对其进行并行化，以分割所有CPU内核的工作负载。

如果您需要在之后将这些补丁提供给Tensorflow图，那么首先要做的是首先在CPU上生成补丁（使用您喜欢的任何输入管道），批处理它们和< em>然后将它们提供给GPU进行图形计算。