mindspore.nn.GetNextSingleOp
- class mindspore.nn.GetNextSingleOp(dataset_types, dataset_shapes, queue_name)[source]
Cell to run for getting the next operation.
For detailed information, refer to mindspore.ops.GetNext.
- Parameters
dataset_types (list[
mindspore.dtype
]) – The types of dataset.queue_name (str) – Queue name to fetch the data.
- Outputs:
tuple[Tensor], the data get from Dataset.
- Supported Platforms:
Ascend
GPU
Examples
>>> import mindspore >>> from mindspore import ops, nn >>> from mindspore import dataset as ds >>> from mindspore.common import dtype as mstype >>> >>> data_path = "/path/to/MNIST_Data/train/" >>> train_dataset = ds.MnistDataset(data_path, num_samples=10) >>> dataset_helper = mindspore.DatasetHelper(train_dataset, dataset_sink_mode=True) >>> dataset = dataset_helper.iter.dataset >>> dataset_types, dataset_shapes = dataset_helper.types_shapes() >>> queue_name = dataset.__transfer_dataset__.queue_name >>> get_next_single_op_net = nn.GetNextSingleOp(dataset_types, dataset_shapes, queue_name) >>> data, label = get_next_single_op_net() >>> relu = ops.ReLU() >>> result = relu(data.astype(mstype.float32)) >>> print(result.shape) (28, 28, 1)