TensorFlow函數(shù):tf.layers.InputSpec

2018-07-12 13:58 更新

tf.layers.InputSpec函數(shù)

InputSpec類

別名:

  • 類 tf.keras.layers.InputSpec
  • 類 tf.layers.InputSpec

定義在:tensorflow/python/layers/base.py.

指定圖層的每個(gè)輸入的ndim,dtype和形狀.

每個(gè)層都應(yīng)公開(如果適用)一個(gè)input_spec屬性:InputSpec的實(shí)例列表(每個(gè)輸入張量一個(gè)).

形狀中的“None”條目與任何維度兼容,None形狀與任何形狀兼容.

參數(shù):

  • dtype:輸入的預(yù)期DataType.
  • shape:形狀元組,輸入的預(yù)期形狀(未經(jīng)檢查的軸可能包括無).
  • ndim:整數(shù),輸入的預(yù)期秩.
  • max_ndim:整數(shù),輸入的最大秩.
  • min_ndim:整數(shù),輸入的最小秩.
  • axes:字典將整數(shù)軸映射到特定的維值.

方法

__init__

__init__(
    dtype=None,
    shape=None,
    ndim=None,
    max_ndim=None,
    min_ndim=None,
    axes=None
)

初始化自我.

以上內(nèi)容是否對(duì)您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號(hào)
微信公眾號(hào)

編程獅公眾號(hào)