reference/generated/tensordict.TensorDictBase.reduce

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

TensorDictBase.reduce

TensorDictBase.reduce(dst, op=None, async_op=False, return_premature=False, group=None) → None

Reduces the tensordict across all machines.

Only the process with rank dst is going to receive the final result.

Every tensordict class and every tensorclass has this method. For the other methods, see Methods by task.