PyTorch、NCNN、CV::Mat三者张量的shape

一、PyTorch

在 PyTorch 中,张量(Tensor)的形状通常按照 (N, C, H, W) 的顺序排列,其中:

N 是批量大小(batch size)
C 是通道数(channel number)
H 是高度(height)
W 是宽度(width)

例如,如果你有一个形状为 (32, 3, 64, 64) 的张量,那么你有 32 个图像,每个图像有 3 个通道(例如 RGB),每个通道的高度和宽度都是 64。
注意,这是一种常见的约定,但并非所有的 PyTorch 代码都必须遵循这种约定。有些代码可能会使用不同的顺序,例如 (N, H, W, C)。你应该总是查看你正在使用的具体代码或数据集,以确定它使用的确切顺序。

二、NCNN

NCNN: 在 NCNN 中,张量的形状通常按照 (w, h, c) 的顺序排列,其中 w 是宽度,h 是高度,c 是通道数。这是因为 NCNN 主要用于计算机视觉任务,这种排列方式符合图像数据的常见布局。

三、CV::Mat

CV::Mat: 在 CV 中,张量的形状通常按照 (h, w, c) 的顺序排列,其中 w 是宽度,h 是高度,c 是通道数。

相关推荐

  1. PyTorch、NCNN、CV::Mat三者shape

    2024-02-05 15:10:08       42 阅读
  2. 【PyTorch】(Tensor)生成

    2024-02-05 15:10:08       48 阅读
  3. 标量、向、矩阵和区别?

    2024-02-05 15:10:08       78 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-02-05 15:10:08       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-02-05 15:10:08       101 阅读
  3. 在Django里面运行非项目文件

    2024-02-05 15:10:08       82 阅读
  4. Python语言-面向对象

    2024-02-05 15:10:08       91 阅读

热门阅读

  1. createvm

    2024-02-05 15:10:08       48 阅读
  2. rust ethers-rs 签名与solidity验证签名例子

    2024-02-05 15:10:08       44 阅读
  3. 力扣刷题-27.移除元素

    2024-02-05 15:10:08       52 阅读
  4. Ubuntu文件系统结构

    2024-02-05 15:10:08       54 阅读
  5. Rust个人学习之Rust国内镜像源

    2024-02-05 15:10:08       42 阅读
  6. 【MySQL】-10 MySQL 存储过程

    2024-02-05 15:10:08       39 阅读
  7. MySQL的存储过程

    2024-02-05 15:10:08       41 阅读
  8. mysql 删除分区表数据

    2024-02-05 15:10:08       50 阅读
  9. MVVM下的窗口关闭

    2024-02-05 15:10:08       51 阅读
  10. ChatGPT高效提问—基础知识(NLP)

    2024-02-05 15:10:08       50 阅读