diff --git a/index.bs b/index.bs index f43cc4e3..07a5d582 100644 --- a/index.bs +++ b/index.bs @@ -1158,7 +1158,7 @@ Normalize the tensor values of input features across the batch dimension using [ dictionary MLBatchNormalizationOptions { MLOperand scale; MLOperand bias; - long axis = 1; + unsigned long axis = 1; float epsilon = 1e-5; MLActivation activation; }; @@ -1176,7 +1176,7 @@ partial interface MLGraphBuilder { - *options*: an optional {{MLBatchNormalizationOptions}}. The optional parameters of the operation. - *scale*: an {{MLOperand}}. The 1-D tensor of the scaling values whose length is equal to the size of the input dimension denoted by *options.axis*. - *bias*: an {{MLOperand}}. The 1-D tensor of the bias values whose length is equal to the size of the input dimension denoted by *options.axis*. - - *axis*: a {{long}} scalar. The index to the feature count dimension of the input shape for which the mean and variance values are. When it's not specified, the default value is 1. + - *axis*: an {{unsigned long}} scalar. The index to the feature count dimension of the input shape for which the mean and variance values are. Its value must be in the range [0, N-1] where N is the rank of input tensor. When it's not specified, the default value is 1. - *epsilon*: a {{float}} scalar. A small value to prevent computational error due to divide-by-zero. The default value is 0.00001 when not specified. - *activation*: an {{MLActivation}}. The optional activation function that immediately follows the normalization operation. @@ -1256,16 +1256,14 @@ partial interface MLGraphBuilder { Concatenates the input tensors along a given axis.