mindspore.nn.CentralCrop

class mindspore.nn.CentralCrop(central_fraction)[源代码]

根据指定比例裁剪出图像的中心区域。

参数：

central_fraction (float) - 裁剪比例，必须是float，并且在范围(0.0, 1.0]内。

输入：

image (Tensor) - shape为 \((C, H, W)\) 的三维Tensor，或shape为 \((N,C,H,W)\) 的四维Tensor。

输出：

Tensor，基于输入的三维或四维的float Tensor。

异常：

TypeError - central_fraction 不是float。
ValueError - central_fraction 不在范围(0.0, 1.0]内。

支持平台：

Ascend GPU CPU

样例：

>>> net = nn.CentralCrop(central_fraction=0.5)
>>> image = Tensor(np.random.random((4, 3, 4, 4)), mindspore.float32)
>>> output = net(image)
>>> print(output.shape)
(4, 3, 2, 2)