我正在尝试将Pytorch中的一些代码改编为Tensorflow,尽管我对后者不太熟悉。
我在旧代码中使用expand_as
。Tensorflow中有类似的函数吗?
您正在寻找在TF 1.9中引入的tf.broadcast_to
(最初名称为tf.contrib.framework.broadcast_to
(。