rand函数参数的作用,代表什么

torch.rand(256, 2048, 13, 13)
256, 2048, 13, 13分别代表着什么?

torch.rand(2)是生成一个有2个元素的向量
torch.rand(2,3)是生成一个2行3列的矩阵
就类比一下好了。
torch.rand(256,2048,13,13)是生成一个256行,2048列,13高,13第四维度长度的四维向量
规范点说torch.rand(*sizes)返回一个张量,包含了从区间[0, 1)的均匀分布中抽取的一组随机数。张量的形状由参数sizes定义。

第四个参数不代表什么,就是四阶张量。零阶是标量,一阶是向量,二阶是矩阵,三阶勉强理解为深度,四阶以上就只能叫多少阶的张量了。