三维数组:通常用于表示单个图像,如(高度,宽度,通道数)。
四维数组:用于表示图像批次,如(批次大小,高度,宽度,通道数)。
五维数组:用于表示视频或序列数据,如(批次大小,时间步长,高度,宽度,通道数)。
NumPy:主要用于数学和科学计算,处理数组和矩阵运算。
PyTorch:主要用于深度学习,支持自动微分,可以在GPU上运行。
NumPy 更适合数据分析和快速数学运算,而 PyTorch 更适合构建和训练神经网络。
#资料来自李沐大神