tf.nn.moments(inputs, moments_dims, name='moments')用pytorch改写

在PyTorch中，可以使用torch.mean()和torch.var()函数来计算均值和方差，相当于tf.nn.moments()函数的作用。因此，可以将tf.nn.moments()函数转换为以下PyTorch代码：

import torch

def moments(inputs, moments_dims, name='moments'):
    mean = torch.mean(inputs, dim=moments_dims)
    var = torch.var(inputs, dim=moments_dims)
    return mean, var

其中，inputs表示输入张量，moments_dims表示需要计算均值和方差的维度，name表示命名空间。返回值为均值和方差。