Add utility method to convert TimeStepBatch to torch tensors

rlworkgroup / garage

A toolkit for reproducible reinforcement learning research.

MIT License

1.86k stars 310 forks source link

Open maliesa96 opened 3 years ago

maliesa96 commented 3 years ago

A method in TimeStepBatch to convert the batch to torch tensors (ie. TimeStepBatch.torch()) would be useful.