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.