解锁式学习
1151人加入学习
(0人评价)
Pytorch学习
价格 免费
承诺服务
该课程属于 1446-肖同学-算法方向-计算机视觉-就业:否 请加入后再学习

dim、keepdim

当我们指定维度之后返回的最大值和最小值,会自动消减一个维度,如果对一个二维数组取最大值之后,还想保持它的维度是两个,那么我们可以设置keepdim=True

 

[展开全文]