获取自定义keras层中当前输入的batch_size维

问题描述

我正在实现自定义Keras层,我需要知道当前输入批次的批次大小才能进行计算。

当然,当我尝试获取输入形状时,形状为“无”。

有没有办法让我以int形式获得形状? 我可以从中得到张量的形状

keras.backend.shape 

但是我无法以可在图层内部使用的形式获取值。有人知道如何在TF.2.0中做到这一点吗? 预先感谢。

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)