ray.rllib.policy.sample_batch.SampleBatch.rows

SampleBatch.rows() → Iterator[Dict[str, numpy.array | jnp.ndarray | tf.Tensor | torch.Tensor]]

返回一个数据行的迭代器，即包含列值的字典。

注意，如果 seq_lens 在 self 中被设置，我们在行中将其设置为 1。

生成器:

本次迭代中行的列值。

from ray.rllib.policy.sample_batch import SampleBatch
batch = SampleBatch({
   "a": [1, 2, 3],
   "b": [4, 5, 6],
   "seq_lens": [1, 2]
})
for row in batch.rows():
    print(row)

{"a": 1, "b": 4, "seq_lens": 1}
{"a": 2, "b": 5, "seq_lens": 1}
{"a": 3, "b": 6, "seq_lens": 1}