ray.experimental.TensorTransportManager.recv_multiple_tensors#

abstract TensorTransportManager.recv_multiple_tensors(obj_id: str, tensor_transport_metadata: TensorTransportMetadata, communicator_metadata: CommunicatorMetadata, target_buffers: List[Any] | None = None) → List[Any][source]#

Receive multiple tensors from the source actor. This is called on the destination actor.

Parameters:

  • obj_id – The object ID for related GPU object.

  • tensor_transport_metadata – The tensor transport metadata for the GPU object.

  • communicator_metadata – The communicator metadata for the send/recv operation.

  • target_buffers – Pre-allocated buffers to receive the tensors into if possible.

Returns:

The received tensors.

Return type:

List[Any]