reference/generated/tensordict.TensorDictBase.arecv

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

TensorDictBase.arecv

async TensorDictBase.arecv(src: TensorDictPipe, *, device: device | str | None = None) → TensorDictBase

Receives content into this tensordict from a UCXX pipe (async).

Parameters:

src (TensorDictPipe) – the pipe to receive from.

Keyword Arguments:

device – device for storage allocation on first receive.

Returns:

The received TensorDict (may be self for in-place updates).

See also

recv() for the synchronous variant and torch.distributed-based transport.

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