>>> 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 则是长和宽。
>>> a = np.array([[1,2,3],[4,5,6]])
>>> np.size(a)
6
>>> np.size(a,1)
3
>>> np.size(a,0)
2
本文标题:size函数(numpy和pytorch不同)
本文链接:https://www.haomeiwen.com/subject/wulqfdtx.html
网友评论