mxnet.gluon.data.vision.transforms.ToTensor的作用

mxnet.gluon.data.vision.transforms.ToTensor的作用：

• 将图像或成批图像转换为张量NDArray。
• 将[0,255]范围内的图像（NDArray of shape (H x W x C)）转换为[0,1]范围内的float32张量（NDArray of shape (C x H x W)）。
• 例子：

from mxnet.gluon import data as gdata
from mxnet import nd
import numpy as np
transformer = gdata.vision.transforms.ToTensor()
#从均匀分布中随机抽取样本,样品均匀分布在半开区间[低，高](含低，不含高)。
image = nd.random.uniform(0, 255, (4, 2, 3)).astype(dtype=np.uint8)
print(image)
print(transformer(image))

运行结果：


其中，均匀分布uniform用法如下：