forked from mindspore-Ecosystem/mindspore
13 lines
446 B
ReStructuredText
13 lines
446 B
ReStructuredText
mindspore.dataset.vision.HWC2CHW
|
|
================================
|
|
|
|
.. py:class:: mindspore.dataset.vision.HWC2CHW()
|
|
|
|
将输入图像的shape从 <H, W, C> 转换为 <C, H, W>。
|
|
如果输入图像的shape为 <H, W> ，图像将保持不变。
|
|
|
|
.. note:: 此操作支持通过 Offload 在 Ascend 或 GPU 平台上运行。
|
|
|
|
异常：
|
|
- **RuntimeError** - 如果输入图像的shape不是 <H, W> 或 <H, W, C>。
|