ray.experimental.TensorTransportManager.send_multiple_tensors#
- abstract TensorTransportManager.send_multiple_tensors(tensors: List[Any], tensor_transport_metadata: TensorTransportMetadata, communicator_metadata: CommunicatorMetadata)[source]#
Send multiple tensors or jax arrays to the destination actor. This is called on the source actor.
- Parameters:
tensors – The tensors or jax arrays to send.
tensor_transport_metadata – The tensor transport metadata for the RDT object.
communicator_metadata – The communicator metadata for the send/recv operation.