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

时间:2020-08-19 15:53:24

标签: python tensorflow keras-layer

我正在尝试创建一个具有动态批处理大小的自定义图层(第一个暗号为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?

0 个答案:

没有答案