enum mindietorch.TensorFormat
说明
描述模型输入的数据排布。
成员
成员名称 |
描述 |
|---|---|
mindietorch.TensorFormat.UNKNOWN |
未知的数据格式 |
mindietorch.TensorFormat.NCHW |
[输入批次大小，输入通道数，输入高度，输入宽度] |
mindietorch.TensorFormat.NHWC |
[输入批次大小，输入高度，输入宽度，输入通道数] |
mindietorch.TensorFormat.ND |
[输入批次大小，输入维度] |
mindietorch.TensorFormat.NC1HWC0 |
[输入批次大小，输入通道数1，输入高度，输入宽度，输入通道数0] |
父主题： 数据结构