size函数（numpy和pytorch不同）

>>> a = np.array([[1,2,3],[4,5,6]])

>>> np.size(a)

6

>>> np.size(a,1)

3

>>> np.size(a,0)

2

而对于torch中的张量来说，size()函数得到的 Size 则是长和宽。