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

左手的ㄟ右手 2022-02-21 10:13 232阅读 0赞

获取单输入尺寸，该层只被使用了一次。

import keras
    from keras.layers import Input, LSTM, Dense, Conv2D
    from keras.models import Model
    a = Input(shape=(32, 32, 3))
    b = Input(shape=(64, 64, 3))
    
    conv = Conv2D(16, (3, 3), padding='same')
    conved_a = conv(a)
    
    # 到目前为止只有一个输入，以下可行：
    assert conv.input_shape == (None, 32, 32, 3)

如果该层被使用了两次

import keras
    from keras.layers import Input, LSTM, Dense, Conv2D
    from keras.models import Model
    a = Input(shape=(32, 32, 3))
    b = Input(shape=(64, 64, 3))
    
    conv = Conv2D(16, (3, 3), padding='same')
    conved_a = conv(a)
    
    # 到目前为止只有一个输入，以下可行：
    assert conv.input_shape == (None, 32, 32, 3)
    
    conved_b = conv(b)
    # 现在 `.input_shape` 属性不可行，但是这样可以：
    assert conv.get_input_shape_at(0) == (None, 32, 32, 3)
    assert conv.get_input_shape_at(1) == (None, 64, 64, 3)

如果是输出，只需要改成output就好：

import keras
    from keras.layers import Input, LSTM, Dense, Conv2D
    from keras.models import Model
    a = Input(shape=(32, 32, 3))
    b = Input(shape=(64, 64, 3))
    
    conv = Conv2D(16, (3, 3), padding='same')
    conved_a = conv(a)
    
    # 到目前为止只有一个输入，以下可行：
    assert conv.input_shape == (None, 32, 32, 3)
    
    conved_b = conv(b)
    # 就改了output，当然尺寸我也改了
    assert conv.get_output_shape_at(0) == (None, 32, 32, 16)
    assert conv.get_output_shape_at(1) == (None, 64, 64, 16)

觉得有帮助请点赞，觉得不明白请回复，觉得有问题请礼貌的回复

发表评论取消回复

表情：

评论列表（有 0 条评论，232人围观）

还没有评论，来说两句吧...

相关阅读

相关 keras：构建多输入多输出模型、多输入单输出模型

一个具有两个输入和两个输出的模型：我们试图预测 Twitter 上的一条新闻标题有多少转发和点赞数。模型的主要输入将是新闻标题本身，即一系列词语，但是为了增添趣味，...

我不是女神ヾ/ 2024年04月18日 17:32/ 0 赞/ 43 阅读

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

更多编程教程请到：[菜鸟教程][Link 1] https://www.piaodoo.com/ 友情链接： [高州阳光论坛][Link 2]https://www.hnt

今天药忘吃喽~/ 2022年12月28日 04:04/ 0 赞/ 136 阅读

相关 Keras获取中间层输出

其中，self里面有三个对象：keras compile出来的图，self.model。 model对应的图 self.graph，model所在的会话 self.sess

迷南。/ 2022年12月20日 06:09/ 0 赞/ 131 阅读

相关 Keras 给定输入数据，获取LOSS关于输入的梯度

需求论文需要使用对抗训练 adversarial training，里面需要拿到目标函数的loss关于输入样本的梯度。方法：使用Keras.backbend的fu

女爷i/ 2022年12月19日 13:10/ 0 赞/ 195 阅读

相关卷积核尺寸计算与输入输出尺寸的关系

W：输入图片大小WxW P：padding的像素数 F：卷积核尺寸 S：步长 D：膨胀率 Fo：膨胀后的卷积核尺寸 N：输出图片大小NxN 则有： F 0 =

Dear 丶/ 2022年12月11日 12:19/ 0 赞/ 138 阅读

相关 [keras] 中间层输出

把每个层的输出都拿出来，然后算均方差 K.var(layeroutput, axis=None, keepdims=False): 目前还没找到在运行过程中每一个ste

超、凢脫俗/ 2022年05月13日 10:42/ 0 赞/ 159 阅读

相关 [keras]自定义loss层+接受输入

　loss函数如何接受输入值 keras封装的比较厉害，官网给的例子写的云里雾里，在stackoverflow找到了[答案][Link 1] You can w

逃离我推掉我的手/ 2022年05月13日 01:34/ 0 赞/ 176 阅读

相关 keras 获取某层的输入/输出 tensor 尺寸

获取单输入尺寸，该层只被使用了一次。 import keras from keras.layers import Input, LSTM, Dense, Co

左手的ㄟ右手/ 2022年02月21日 10:13/ 0 赞/ 233 阅读

相关 keras 获取某层输出获取复用层的多次输出

官方文档很全面，搜索功能也很好。但是如果你想单独实现某个功能，根本无从搜寻。于是我写了这个笔记。从功能出发。两个tensor经过一个layer实例会产生两个输出。

た入场券/ 2022年02月21日 09:57/ 0 赞/ 194 阅读

相关 keras:tensor从全连接层输出到卷积层

一. tensor从卷积层输出到全连接层用过keras的都知道，想从卷积层输出tensor到全连接层，只需加一层： model.add(Flatten())

刺骨的言语ヽ痛彻心扉/ 2021年12月21日 02:15/ 0 赞/ 323 阅读