YouTip LogoYouTip

Pytorch Torch Nanmean

# PyTorch torch.nanmean Function * * * [![Image 3: Pytorch torch Reference Manual]( Pytorch torch Reference Manual]( `torch.nanmean` is a function in PyTorch used to calculate the mean while ignoring NaN values. ### Function Definition torch.nanmean(input, dim, keepdim=False, out=None) * * * ## Usage Examples ## Example import torch # Create tensor containing NaN x = torch.tensor([1.0,2.0,float('nan'),4.0,5.0]) # Calculate mean of non-NaN values mean = torch.nanmean(x) print("Non-NaN mean:", mean) Output result: Non-NaN mean: tensor(3.) * * * [![Image 4: Pytorch torch Reference Manual]( Pytorch torch Reference Manual](
← Pytorch Torch NanmedianPytorch Torch Mvlgamma β†’