torch.mean() | 返回平均值 |
torch.sum() | 返回总和 |
torch.prod() | 计算所有元素的积 |
torch.max() | 返回最大值 |
torch.min() | 返回最小值 |
torch.argmax() | 返回最大值排序的索引值 |
torch.argmin() | 返回最小值排序的索引值 |
Pytorch中常见的聚合函数
最新推荐文章于 2024-06-22 18:11:08 发布
![](https://img-home.csdnimg.cn/images/20240711042549.png)
torch.mean() | 返回平均值 |
torch.sum() | 返回总和 |
torch.prod() | 计算所有元素的积 |
torch.max() | 返回最大值 |
torch.min() | 返回最小值 |
torch.argmax() | 返回最大值排序的索引值 |
torch.argmin() | 返回最小值排序的索引值 |