张量流while-loop并行化张量阵列

2024-05-17 16:35:17 发布

您现在位置:Python中文网/ 问答频道 /正文

我不太明白while循环并行化是如何工作的。假设我有一个张量阵列,有10个相同形状的张量。现在假设循环体中前5个张量的计算与其余5个张量的计算无关。TensorFlow会让这两个并行运行吗?另外,如果我使用张量而不是张量数组,并使用scatter_update对其进行更新,它会在backprop期间正确通过渐变吗?在


Tags: tensorflowupdate数组形状whilescatterbackprop化是
1条回答
网友
1楼 · 发布于 2024-05-17 16:35:17

您可能应该获得前5个迭代和第2个5个迭代的并行执行。我可以肯定地说,如果你提供一个代码示例。在

相关问题 更多 >