TensorFlow函數(shù)：tf.unstack

tf.unstack函數(shù)

tf.unstack(
    value,
    num=None,
    axis=0,
    name='unstack'
)

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

參見(jiàn)指南：張量變換>分割和連接

將秩為 R 的張量的給定維度出棧為秩為 (R-1) 的張量.

通過(guò)沿 axis 維度將 num 張量從 value 中分離出來(lái).如果沒(méi)有指定 num(默認(rèn)值),則從 value 的形狀推斷.如果 value.shape[axis] 不知道,則引發(fā) ValueError.

例如,給定一個(gè)具有形狀 (A, B, C, D) 的張量.

• 如果 axis == 0,那么 output 中的第 i 個(gè)張量就是切片 value[i, :, :, :],并且 output 中的每個(gè)張量都具有形狀 (B, C, D).(請(qǐng)注意,出棧的維度已經(jīng)消失,不像split).
  
• 如果 axis == 1,那么 output 中的第 i 個(gè)張量就是切片 value[:, i, :, :],并且 output 中的每個(gè)張量都具有形狀 (A, C, D).
  

這與堆棧(stack.)相反,numpy 相當(dāng)于：

tf.unstack(x, n) = np.unstack(x)

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

• value：一個(gè)要出棧的秩 R > 0 的 Tensor.
• num：一個(gè) int,維度 axis 的長(zhǎng)度,如果為 None(默認(rèn)值),則自動(dòng)推斷.
• axis：一個(gè) int,沿著這個(gè)軸出棧,默認(rèn)為第一維,負(fù)值環(huán)繞,所以有效范圍是 [-R, R).
• name：操作的名稱(可選).

函數(shù)返回值：

tf.unstac k函數(shù)從 value 中出棧的 Tensor 對(duì)象列表.

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

• ValueError：如果 num 沒(méi)有指定并且無(wú)法推斷.
• ValueError：如果 axis 超出范圍 [-R,R).