tf.keras.Input

概述

该函数用于:实例化keras张量,向模型输入数据,需要指定 数据的 形状

一般搭配 tf.keras.Model() 函数式api使用,作为最开始的输入

语法

tf.keras.Input(
    shape=None,
    batch_size=None,
    name=None,
    dtype=None,
    sparse=False,
    tensor=None,
    ragged=False,
    **kwargs
)
参数说明:
  • shape:整数或tuple类型,数据形状。可以是None,表示未知
  • batch_size:静态批量大小
  • name:名称,可选,应唯一
  • dtype:字符串类型,表示数据类型。如:float32,float64,int32
返回值:

tensor

例子

inp=tf.keras.Input(shape=(img_width,img_width,3))

更多参见:
https://www.malaoshi.top/show_1EF4TOg6mzFU.html


原文出处:/show_1EF4TfThprkp.html