具有动态批次大小的tensorflow 2自定义层

编程问答 2022-08-16

问题描述

我正在尝试创建一个具有动态批处理大小的自定义图层（第一个暗号为None），该图层将根据先前的输入并输出到该图层（来自先前的批处理）来计算该图层的当前输出。
为简单起见，假设我正在尝试计算以下内容：

def call(self,current_inputs):
  current_output = current_inputs + self.prevIoUs_inputs + self.prevIoUs_outputs
  self.prevIoUs_inputs = current_inputs  #will be used for the next call
  self.prevIoUs_outputs = current_output
  return current_output

如果我具有动态批处理大小，如何初始化prevIoUs_inputs和prevIoUs_outputs？

解决方法

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

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

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

keras-layerpythontensorflowtensorflowtensorflow