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

Question

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

我在旧代码中使用expand_as。 Tensorflow中可以使用任何类似的功能吗？

Answer 1

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