系统信息

• 我是否编写了自定义代码（而不是使用TensorFlow中提供的常用示例脚本）：没有
• 操作系统平台和分发版（例如Linux Ubuntu 16.04）：Linux操作系统Ubuntu 18.04
• 移动的设备（例如iPhone 8、Pixel 2、Samsung Galaxy），如果问题发生在移动设备上：不适用
• TensorFlow安装自（源代码或二进制文件）：二进制的
• TensorFlow版本（使用下面的命令）：2.1.0
• Python版本：3.7.6
• Bazel版本（如果从源代码编译）：不适用
• GCC/编译器版本（如果从源代码编译）：不适用
• CUDA/cuDNN版本：不适用
• GPU型号和内存：不适用
  描述当前行为

当axes=[]和x为复数时，tf.nn.weighted_moments输出NaN

描述预期行为

输出中不应有Nan

重现问题的独立代码

import tensorflow as tf
import numpy as np
tf.nn.weighted_moments(x=np.array([1e+38+0.j, 2e+38+0.j], dtype=np.complex64), frequency_weights=10, axes=[])

输出量：

(<tf.Tensor: shape=(2,), dtype=complex64, numpy=array([inf+nanj, inf+nanj], dtype=complex64)>, <tf.Tensor: shape=(2,), dtype=complex64, numpy=array([inf+nanj, inf+nanj], dtype=complex64)>)