PyTorch torch.nanmean 函数

PyTorch torch.nanmean 函数 Pytorch torch 参考手册 torch.nanmean 是 PyTorch 中用于计算忽略 NaN 值的平均值的函数。函数定义 torch.nanmean(input, dim, keepdim=False, out=None) 使用示例实例 [mycode4 type='python'] import torch # 创建包含 N..

更新于 2026/7/16

教程详情页默认 noindex。

PyTorch torch.nanmean 函数

Pytorch torch 参考手册

torch.nanmean 是 PyTorch 中用于计算忽略 NaN 值的平均值的函数。

函数定义

torch.nanmean(input, dim, keepdim=False, out=None)

使用示例

实例

import torch

# 创建包含 NaN 的张量
x = torch.tensor([1.0, 2.0, float('nan'), 4.0, 5.0])

# 计算非 NaN 值的平均值
mean = torch.nanmean(x)
print("非 NaN 均值:", mean)

输出结果为：

非 NaN 均值: tensor(3.)

Pytorch torch 参考手册