t = torch.rand(3,48)
t.mean(dim=0,keepdim=True).shape # torch.Size([1, 48])
t = torch.rand(3,100,100,100)
t.mean(dim=(0,2,3),keepdim=True).shape # torch.Size([1, 100, 1, 1])
均值
©著作权归作者所有,转载或内容合作请联系作者
- 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
- 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
- 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
推荐阅读更多精彩内容
- (1)、算数平均值法求权重 步骤: 判断矩阵按列求和,得到新矩阵a_axis_0_sum 把判断矩阵中的每一个数都...
- 从输入任意个整型数,统计其中的负数个数并求所有非负数的平均值,结果保留一位小数,如果没有非负数,则平均值为0 本题...