分类：tensor尺寸

2020
10-07

keras 获取某层的输入/输出 tensor 尺寸操作

获取单输入尺寸，该层只被使用了一次。importkerasfromkeras.layersimportInput,LSTM,Dense,Conv2Dfromkeras.modelsimportModela=Input(shape=(32,32,3))b=Input(shape=(64,64,3))conv=Conv2D(16,(3,3),padding='same')conved_a=conv(a)#到目前为止只有一个输入，以下可行：assertconv.input_shape==(None,32,32,3)如果该层被使用了两次importkerasfromkeras.layersimportInput,LSTM,Dens... 继续阅读 >