gpt4 book ai didi

python - 如何改变Python中的张量形状?

转载 作者:行者123 更新时间:2023-12-01 08:17:57 24 4
gpt4 key购买 nike

假设在 Tensorflow 中,我有一个形状为 [3,4,4] 的张量,该张量尚不清楚。如何将其形状更改为 [3,4,4,1]
我尝试使用 reshape 函数,但由于我不明确知道张量形状,因此无法直接使用它。

最佳答案

给定形状 [3,4,4],它可以 reshape 为 [3,4,4, 1]

tf.reshape(t, [3,4,4, 1]) # by explicitly specifying the shape
tf.expandims(t, 3) # implicitly we expand the dimension 3 with size 1

关于python - 如何改变Python中的张量形状?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54879128/

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