根据批号获取批输入张量

Question

如何在批处理结束时触发的回调中访问给定批次的输入数据？我可以找出批号，但这只是一个整数。我想访问输入到输入层的输入张量，换句话说是批量的训练数据。这取决于给定的时期，因为数据随着每个新纪元而混乱。

Answer 1

training loop没有传递输入，因此您无法从参数中获取。一种选择是根据提供的batch_index：

自行访问数据和计算数据点

def on_batch_end(self, batch, logs={}):
  dpoints = data[batch * batch_size:(batch + 1) * batch_size]

如果您的数据不是全局数据，则可以在创建回调时将其设置为类字段。