oneflow.comm.recv¶
-
oneflow.comm.
recv
()¶ Receives a tensor synchronously.
All(send_meta is False) or none of shape, dtype and device should have value.
- Parameters
src (int, optional) – Source rank. Will receive from any process if unspecified.
shape (optional) – output tensor shape.
dataType (optional) – output tensor data type.
device (optional) – output tensor device.
out (Tensor, optional) – Tensor to fill with received data.
- Returns
if out is None, return received tensor. otherwise got data from out self without return.