receive【免费下载链接】tensorflowAscend TensorFlow Adapter项目地址: https://gitcode.com/cann/tensorflow功能说明提供group内点对点通信数据的receive功能。函数原型def receive(shape, data_type, sr_tag, src_rank, grouphccl_world_group)参数说明参数名输入/输出描述shape输入接收tensor的shape。data_type输入接收数据的数据类型。针对Ascend 950PR/Ascend 950DT支持数据类型int8、uint8、int16、uint16、int32、uint32、int64、uint64、float16、float32、float64、bfp16。针对Atlas A3 训练系列产品/Atlas A3 推理系列产品支持数据类型int8、uint8、int16、uint16、int32、uint32、int64、uint64、float16、float32、float64、bfp16。针对Atlas A2 训练系列产品/Atlas A2 推理系列产品支持数据类型int8、uint8、int16、uint16、int32、uint32、int64、uint64、float16、float32、float64、bfp16。针对Atlas 训练系列产品支持数据类型int8、uint8、 int16、uint16、int32、uint32、int64、uint64、float16、float32、float64。针对Atlas 300I Duo 推理卡支持数据类型int8、uint8、int16、uint16、int32、uint32、int64、uint64、float16、float32、float64。sr_tag输入消息标签相同sr_tag的send/recv对可以收发数据int类型。src_rank输入接收数据的源节点该rank是group中的rank idint类型。group输入String类型最大长度为128字节含结束符。group名称可以为用户自定义group或者hccl_world_group。返回值进行receive操作之后从对端接收到的tensor。约束说明调用该接口的rank必须在当前接口入参group定义的范围内不在此范围内的rank调用该接口会失败。send和receive必须配对使用即调用send接口后需要等到与之配对的receive接口接收数据后才可以进行下一个接口调用。调用示例from npu_bridge.hccl import hccl_ops tensor tf.random_uniform((1, 3), minval1, maxval10, dtypetf.float32) sr_tag 0 src_rank 0 tensor hccl_ops.receive(tensor.shape, tensor.dtype, sr_tag, src_rank)【免费下载链接】tensorflowAscend TensorFlow Adapter项目地址: https://gitcode.com/cann/tensorflow创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考