mindspore.dataset.dataloader.default_convert

mindspore.dataset.dataloader.default_convert(data)

当在 DataLoader 中禁用批处理时，默认使用此函数将NumPy数组类型的元素转换为 mindspore.Tensor 。

• 如果输入是NumPy数组且其数据类型不为 str、bytes 或 object，则将其转换为 mindspore.Tensor；

• 如果输入是NumPy数值或布尔类型标量，则将其转换为 mindspore.Tensor；

• 如果输入是映射（ Mapping ）类型，则保持所有键不变，递归调用此函数转换各个键对应的值；

• 如果输入是序列（ Sequence ）类型，则递归调用此函数转换各个位置的元素；

• 否则，保持不变。

参数：

• data (Any) - 要转换的单个数据。

返回：

Any ，转换后的数据。

样例：

>>> import numpy as np
>>> from mindspore.dataset.dataloader import default_convert
>>>
>>> default_convert(np.array([0, 1, 2]))
Tensor(shape=[3], dtype=Int64, value= [0, 1, 2])
>>>
>>> default_convert(np.int32(0))
Tensor(shape=[], dtype=Int32, value= 0)
>>>
>>> default_convert({"data": np.array([0, 1, 2])})
{'data': Tensor(shape=[3], dtype=Int64, value= [0, 1, 2])}
>>>
>>> default_convert([np.array([0, 1, 2]), np.array([3, 4, 5])])
[Tensor(shape=[3], dtype=Int64, value= [0, 1, 2]), Tensor(shape=[3], dtype=Int64, value= [3, 4, 5])]
>>>
>>> default_convert(np.array(["text"]))
array(['text'], dtype='<U4')