tensorflow - Tensorflow 中哪个函数类似于 Pytorch 中的 expand_as

Question

我正在尝试将一些代码从 Pytorch 改编为 Tensorflow，尽管我对后者不太熟悉。

我expand_as在我的旧代码中使用。Tensorflow 中是否有类似的功能？

标签： tensorflowreshapepytorchtensor

Answer 1

您正在寻找tf.broadcast_to已在 TF 1.9 中引入的（最初在名称下tf.contrib.framework.broadcast_to）。