torch.rand(256, 2048, 13, 13)
256, 2048, 13, 13分别代表着什么?
2条回答 默认 最新
八云黧 2021-07-29 10:33关注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定义。本回答被题主选为最佳回答 , 对您是否有帮助呢?评论 打赏 举报解决 1无用