loveanddestiny的博客

原创 关于numpy和torch.tensor中的size和shape

对于numpy中申请的array变量，size和shape是array类的成员属性： import numpy as np b = np.random.rand(2,3,5) print(b.size)##输出2*3*5=30，即array元素的个数 print(b.shape)##输出(2,3,5)的tuple,即array的维度， print(len(b))##使用len()函数，返回的是array的第一维值 对于torch.tensor ，shape是tensor的成员属性，size是tensor的