繁体   English   中英

如何在python中改变张量形状?

[英]How to change tensor shape in python?

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

给定形状[3,4,4] t ,可以将其重塑为[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

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM