n 维数组(张量)关于轴 axis 的理解

本文将从两个角度来理解 “轴” 的概念,着重阐述 1.2 节中的理解,并借此加深问题一和问题二的理解。
一、问题:如何理解 numpy 数组在轴上的 sum 操作
二、问题:torch 张量中的维度 dim 也是如此

一、问题:如何理解 numpy 数组在轴上的 sum 操作

a = np.ones((2, 3, 4), dtype=int)
a.sum(axis=<

你可能感兴趣的:(python,numpy,pytorch)