張形狀態斷言和聲明

Question

我使用tf.strided_slice從1d張量中獲取一個值。不幸的是，推斷的形狀是？我怎樣才能斷言/宣稱它已經形成[1]？

P.S.我使用了重塑，但在某些情況下可能會影響性能

Answer 1

使用x.set_shape()可提供有關此張量形狀的附加信息，該信息無法單獨從圖表中推斷出來。

你可以從FAQ的更多信息：

的tf.Tensor.set_shape方法更新張量 對象的靜態形狀，它通常用於提供額外的形狀 信息時，這不可能直接推斷。它並不改變張量的動態形狀。