利用python随机生成一个张量
a=torch.randn(2,3,1)
输出:
基于上述数据可以得到以下观察,首先确定这是一个三维向量,这个可以根据最左边方括号([ )的数量来确定。
那么它的形状必然是(a,b,c)的形式。
看左边的第一个方括号([ )里面包含了几个元素,这里的元素是以([ ])为一组,因此a=2;再看第二个方括号([ )里包含了几个元素,这里的元素是以([ ])为一组,因此b=3;最后看第三个方括号([ )里面的元素个数,由于这是最后一维数据了,因此直接数这组括号里面的元素个数,因此c=1.因此a.size()=([2,3,1])
为了更好的理解这个概念,我们利用python再随机生成一组数据
b=torch.randn(2,1,3)
输出:
仍然按照上述方法进行判断,最左边的方括号里面包含2组次左方括号,a=2;次左方括号里面包含一组次次左方括号,b=1;最后一组方括号里面包含3个元素,c=3.因此,b.size()=([2,1,3])