def moments(
x,
axes,
shift=None, # pylint: disable=unused-argument
name=None,
keep_dims=False)
x: 形如:[batch_size, height, width, kernels]
axes: 求解的维度,如 [1,2,3]
keep_dims: 是否保持维度
函数本身是用来计算均值和方差,也是 BN 中主要使用两个 api 之一:
def moments(
x,
axes,
shift=None, # pylint: disable=unused-argument
name=None,
keep_dims=False)
x: 形如:[batch_size, height, width, kernels]
axes: 求解的维度,如 [1,2,3]
keep_dims: 是否保持维度
函数本身是用来计算均值和方差,也是 BN 中主要使用两个 api 之一: