torch.mean() | 返回平均值 |
torch.sum() | 返回总和 |
torch.prod() | 计算所有元素的积 |
torch.max() | 返回最大值 |
torch.min() | 返回最小值 |
torch.argmax() | 返回最大值排序的索引值 |
torch.argmin() | 返回最小值排序的索引值 |
Pytorch中常见的聚合函数
最新推荐文章于 2024-05-16 19:14:45 发布
torch.mean() | 返回平均值 |
torch.sum() | 返回总和 |
torch.prod() | 计算所有元素的积 |
torch.max() | 返回最大值 |
torch.min() | 返回最小值 |
torch.argmax() | 返回最大值排序的索引值 |
torch.argmin() | 返回最小值排序的索引值 |