reference/generated/tensordict._ucxx.send_tensordict

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

send_tensordict

async tensordict._ucxx.send_tensordict(endpoint, td: TensorDictBase, *, consolidated: bool = True) → None

Send a TensorDict over a UCXX endpoint (stateless, always sends metadata).

Warning

send_tensordict is experimental: its API can change in a minor release without a deprecation cycle.

Parameters:

• endpoint – a UCXX endpoint object.

• td – the TensorDict to send.

Keyword Arguments:

consolidated (bool) – if True, consolidates into a single buffer before sending. Defaults to True.