gpt4 book ai didi

python - 在 Tensorflow 中,最后一个维度如何使用 tf.gather()?

转载 作者:太空狗 更新时间:2023-10-29 17:52:01 26 4
gpt4 key购买 nike

我正在尝试根据层之间的部分连接的最后一个维度来收集张量的切片。因为输出的tensor的shape是[batch_size, h, w, depth],我想根据最后一个维度来选择切片,比如

# L is intermediate tensor
partL = L[:, :, :, [0,2,3,8]]

但是,tf.gather(L, [0, 2,3,8]) 似乎只适用于第一个维度(对吧?)谁能告诉我该怎么做?

最佳答案

从 TensorFlow 1.3 开始,tf.gather 有一个 axis 参数,因此不再需要此处的各种解决方法。

https://www.tensorflow.org/versions/r1.3/api_docs/python/tf/gather https://github.com/tensorflow/tensorflow/issues/11223

关于python - 在 Tensorflow 中,最后一个维度如何使用 tf.gather()?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/36764791/

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