gpt4 book ai didi

python - Theano 的 dimshuffle 的 Tensorflow 等价物

转载 作者:太空宇宙 更新时间:2023-11-03 16:05:57 26 4
gpt4 key购买 nike

如何找到 Theano 的 dimshuffle 函数的 Tensorflow 等效项。

mask = mask.dimshuffle(0, 1, "x")

我正在这样做,我尝试了很多方法但找不到。谢谢:)

最佳答案

您可以使用tf.transpose :

 mask = tf.transpose(mask, perm=[0, 1])

关于python - Theano 的 dimshuffle 的 Tensorflow 等价物,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/39790829/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com