[英]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.