def append(self, tensors):
position = tf.mod(self.index, self.capacity)
append_ops = [self.buffers[key][position].assign(tensor) for key, tensor in zip(self.buffers, tensors)]
with tf.control_dependencies(append_ops):
inc_index_op = self.index.assign_add(1)
return inc_index_op
评论列表
文章目录