libo99999的博客

A.shape[0]和A.shape[1]分别代表矩阵A的行数和列数。

0 含义：0代表行，最后结果是将每列的所有元素求和，结果剩下一行。最初看axis=0和axis=1时不懂，现在懂了，分享一下！说一下torch.size（[4]）含义是1*4=4。代表列，最后结果是将每行的所有元素求和，结果剩下一列。torch.size（[5]）含义是5*1=5。这是矩阵a(或者叫张量a)，条件下求和赋值给变量。