假设在给2D-CNN输入之后,作为第i层的输出,我有一个Output Shape: (None, 1, 3, 1).我还有另一个关于Shape (None, 50, 27, 1)的输入,我想要将列1-9的每个元素与(None,1,3,1)的第一个元素相乘,将列10-18的每个元素与(None,1,3,1)的第二个元素相乘,并将列19-27的每个元素与(None,1,3,1)的第三个元素相乘.总而言之,我想做的事情如下:

enter image description here

然而,使用列表来实现这一点并不困难,我想在我的keras模型中实现这一点.因此,我的输入类型是‘keras.Eng.keras_tensor.KerasT’.你有办法做到这一点吗?

推荐答案

如果您的输入是ab

c = tf.keras.layers.Lambda(
    lambda x: tf.concat(
        [a*b for a, b in zip(
            tf.split(x[1], 3, axis=-2), 
            tf.split(x[0], 3, axis=-2)
            )
        ], axis=-2
        )
    )([a,b])

Python相关问答推荐

Python多处理:当我在一个巨大的pandas数据框架上启动许多进程时,程序就会陷入困境

如何在Windows上用Python提取名称中带有逗号的文件?

PyQt5,如何使每个对象的 colored颜色 不同?'

梯度下降:简化要素集的运行时间比原始要素集长

Stacked bar chart from billrame

无法连接到Keycloat服务器

如何在BeautifulSoup/CSS Select 器中处理regex?

下三角形掩码与seaborn clustermap bug

(Python/Pandas)基于列中非缺失值的子集DataFrame

在Admin中显示从ManyToMany通过模型的筛选结果

具有相同图例 colored颜色 和标签的堆叠子图

python sklearn ValueError:使用序列设置数组元素

干燥化与列姆化的比较

30个非DATETIME天内的累计金额

如何在PYTHON中向单元测试S Side_Effect发送额外参数?

使用pythonminidom过滤XML文件

try 在单个WITH_COLUMNS_SEQ操作中链接表达式时,使用Polars数据帧时出现ComputeError

如何在不遇到IndexError的情况下将基数10的整数转换为基数80?

组颠倒大Pandas 数据帧

是否将列表分割为2?