我正在尝试将一些代码从 Pytorch 改编为 Tensorflow,尽管我对后者不太熟悉。
我在旧代码中使用expand_as
。 Tensorflow中是否有类似的函数可以使用?
最佳答案
您正在寻找tf.broadcast_to
它已在 TF 1.9 中引入(最初名称为 tf.contrib.framework.broadcast_to
)。
关于tensorflow - Tensorflow中哪个函数与Pytorch中的expand_as类似,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/51346819/