Python tf.raw_ops.Squeeze用法及代碼示例

從張量的形狀中刪除大小為 1 的維度。

用法

tf.raw_ops.Squeeze(
    input, axis=[], name=None
)

input 一個Tensor。 input 要擠壓。
axis ints 的可選列表。默認為 [] 。如果指定，僅壓縮列出的尺寸。維度索引從 0 開始。擠壓不是 1 的維度是錯誤的。必須在 [-rank(input), rank(input)) 範圍內。
name 操作的名稱(可選)。

給定一個張量 input ，此操作返回一個相同類型的張量，其中刪除了所有大小為 1 的維度。如果您不想刪除所有尺寸 1 的尺寸，您可以通過指定 axis 來刪除特定的尺寸 1 尺寸。

# 't' is a tensor of shape [1, 2, 1, 3, 1, 1]
shape(squeeze(t)) ==> [2, 3]

或者，要刪除特定尺寸 1 尺寸：

# 't' is a tensor of shape [1, 2, 1, 3, 1, 1]
shape(squeeze(t, [2, 4])) ==> [1, 2, 3, 1]

相關用法

注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.raw_ops.Squeeze。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。