簡體   English   中英

如何在TensorFlow中清除tf.Tensor的形狀信息?

[英]How do I clear the shape information of tf.Tensor in TensorFlow?

假設在訓練網絡期間,我們將所有圖像的大小調整為512 * 512,因此可能有一個名為input:0tf.Tensor input:0 ,形狀為(batch_size, 512, 512, 3) tf.Tensor (batch_size, 512, 512, 3)

但是,在進行預測時,可以將多種尺寸的圖像輸入網絡。 因此張量input:0的形狀input:0應該類似於(batch_size, None, None, 3) ,因為圖像的大小是未知的。

因此,如果我有一個Tensor of shape (batch_size, 512, 512, 3) ,我該如何“重塑”它(batch_size, None, None, 3) 我試過了

inputs=tf.reshape(inputs, (batch_size, tf.shape(inputs)[1], tf.shape(inputs)[2], 3)

但輸出仍然是形狀(batch_size, 512, 512, 3)

我不相信你可以調整/重新調整神經網絡中的權重/偏差項。 但是將圖像輸入的大小調整為512 * 512非常容易。 你考慮過嗎?

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM