TensorFlow函數(shù)教程:tf.nn.static_state_saving_rnn

2019-02-14 14:54 更新

tf.nn.static_state_saving_rnn函數(shù)

別名:

  • tf.contrib.rnn.static_state_saving_rnn
  • tf.nn.static_state_saving_rnn
tf.nn.static_state_saving_rnn(
    cell,
    inputs,
    state_saver,
    state_name,
    sequence_length=None,
    scope=None
)

定義在:tensorflow/python/ops/rnn.py。

為時間截斷的RNN計算接受狀態(tài)保護(hù)程序的RNN。

參數(shù):

  • cell:RNNCell的一個實例。
  • inputs:輸入的長度為T的列表,每個輸入都是一個具有shape [batch_size, input_size]的Tensor。
  • state_saver:一個狀態(tài)保護(hù)程序?qū)ο螅哂蟹椒╯tate和save_state。
  • state_name:Python字符串或字符串元組。與state_saver一起使用的名稱。如果單元格返回狀態(tài)元組(即,cell.state_size是一個元組),則state_name應(yīng)該是與cell.state_size具有相同長度的字符串元組。否則它應(yīng)該是一個單獨的字符串。
  • sequence_length:(可選)int32 / int64向量,大小為[batch_size]。
  • scope:用于創(chuàng)建子圖的VariableScope;默認(rèn)為“rnn”。

返回:

(outputs, state)對,其中:

  • outputs是長度為T的輸出列表(每個輸入一個)
  • state是最終狀態(tài)

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

  • TypeError:如果cell不是RNNCell的實例。
  • ValueError:如果inputs是None或是一個空列表,或者state_name的arity和type與cell.state_size的不匹配。
以上內(nèi)容是否對您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號
微信公眾號

編程獅公眾號