簡體   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