mindspore.dataset.Dataset.device_que

mindspore.dataset.Dataset.device_que(send_epoch_end=True, create_data_info_queue=False, queue_name='')[源代码]

将数据异步传输到Ascend/GPU设备上。

参数：

send_epoch_end (bool, 可选) - 数据发送完成后是否发送结束标识到设备上。默认值： True 。
create_data_info_queue (bool, 可选) - 是否创建一个队列，用于存储每条数据的数据类型和shape。默认值： False ，不创建。
queue_name (str, 可选) - 连接数据处理和模型计算的数据队列的名称。默认值： "" 。

说明

如果设备类型为Ascend，数据的特征将被逐一传输。每次传输的数据大小限制为256MB。

返回：: Dataset，应用了上述操作的新数据集对象。

样例：

>>> import mindspore.dataset as ds
>>> import time
>>>
>>> data = ds.TFRecordDataset('/path/to/TF_FILES', '/path/to/TF_SCHEMA_FILE', shuffle=ds.Shuffle.FILES)
>>> data = data.device_que()
>>> data.send()
>>> time.sleep(0.1)
>>> data.stop_send()