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

2024-04-30 00:59:30 发布

您现在位置：Python中文网/ 问答频道 /正文
1824 3

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


Tags： tensorflowupdate数组形状whilescatterbackprop化是
1条回答
网友
1楼 · 发布于 2024-04-30 00:59:30

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

相关问题 更多 >

    编程相关推荐