numpy.view_as_windows 有 tensorflow 版本吗？答案

【问题标题】：Is there any tensorflow version of numpy.view_as_windows?numpy.view_as_windows 有 tensorflow 版本吗？
【发布时间】：2018-01-23 11:07:13
【问题描述】：

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

例如：view_as_windows(array2d, window_shape, stride)

谢谢

【问题讨论】：

之后您是否需要在 Tensoflow 图中使用这些补丁，或者这只是为了“快速”生成补丁？
我想我希望这只是为了快速生成补丁，因为之后我仍然需要对其进行修改。实际上我是这个 tensorflow 的新手，所以如果你有任何建议，那就太好了：D

标签： python-3.x numpy tensorflow

【解决方案1】：

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

如果您需要做的只是生成图像补丁，Tensorflow（通常是 GPU 加速）不是解决此问题的正确工具，因为实际计算很简单（提取图像），瓶颈将是 GPU 和 CPU 之间的内存传输。

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

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

【讨论】：

谢谢，很好的信息！我将尝试寻找更多关于此的信息。我想我必须为此感到麻木。