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