无法解析导入"tensorflow.math"

Question

我正在VSCode上使用Jupyter Notebooks创建U-Net。以下是我生成错误的代码的快速片段：

# PREPARE U-NET MODEL
from tensorflow.keras import Input, Model
from tensorflow.keras.backend import clear_session
from tensorflow.keras.layers import Activation, Add, BatchNormalization, Concatenate, Convolution2DTranspose, MaxPool2D, SeparableConv2D
from tensorflow.math import reduce_mean

随着新的更新，Pylance现在已集成到Jupyter笔记本中。然而，它给了我一个错误，说tensorflow.math无法解决。很明显，我并没有明确地在TensorFlow中安装数学部分。

给出的具体误差为Pylance(reportMissingImports)。

Answer 1

你能试试吗

from tensorflow._api.v2.math import reduce_mean