TensorFlow函數(shù)：tf.squeeze

tf.squeeze函數(shù)

squeeze(
    input,
    axis=None,
    name=None,
    squeeze_dims=None
)

定義在：tensorflow/python/ops/array_ops.py.

參見指南：張量變換>形狀的確定與改變

從張量形狀中移除大小為1的維度.

給定一個張量 input,該操作返回一個與已經(jīng)移除的所有大小為1的維度具有相同類型的張量.如果您不想刪除所有大小為1的維度,則可以通過指定 axis 來刪除特定的大小為1的維度.

如本例所示：

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

或者,要刪除特定的大小為1的維度：

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

函數(shù)參數(shù)：

• input：A Tensor.該input擠.
• axis：一個可選列表ints.默認為[].如果指定,只能擠壓列出的尺寸.維度索引從0開始.壓縮非1的維度是錯誤的.必須在范圍內(nèi)[-rank(input), rank(input)).
• name：操作的名稱(可選).
• squeeze_dims：現(xiàn)在是軸的已棄用的關(guān)鍵字參數(shù).

函數(shù)返回值：

一Tensor.與.類型相同input.包含與之相同的數(shù)據(jù)input,但刪除了一個或多個尺寸為1的尺寸.

可能引發(fā)的異常：

• ValueError：當兩個squeeze_dims和axis指定.