wangshuide2020
5 years ago
3 changed files with 2 additions and 5 deletions
Unified View
Diff Options
-
+0 -1mindspore/nn/layer/embedding.py
-
+0 -4mindspore/nn/layer/normalization.py
-
+2 -0mindspore/nn/metrics/fbeta.py
+ 0
- 1
mindspore/nn/layer/embedding.py
View File
| @@ -166,7 +166,6 @@ class EmbeddingLookup(Cell): | |||||
| >>> print(result) | >>> print(result) | ||||
| [[[ 0.00856617 0.01039034] | [[[ 0.00856617 0.01039034] | ||||
| [ 0.00196276 -0.00094072]] | [ 0.00196276 -0.00094072]] | ||||
| [[ 0.01279703 0.00078912] | [[ 0.01279703 0.00078912] | ||||
| [ 0.00084863 -0.00742412]]] | [ 0.00084863 -0.00742412]]] | ||||
| """ | """ | ||||
+ 0
- 4
mindspore/nn/layer/normalization.py
View File
| @@ -377,10 +377,8 @@ class BatchNorm2d(_BatchNorm): | |||||
| >>> print(output) | >>> print(output) | ||||
| [[[[171.99915 46.999763 ] | [[[[171.99915 46.999763 ] | ||||
| [116.99941 191.99904 ]] | [116.99941 191.99904 ]] | ||||
| [[ 66.999664 250.99875 ] | [[ 66.999664 250.99875 ] | ||||
| [194.99902 102.99948 ]] | [194.99902 102.99948 ]] | ||||
| [[ 8.999955 210.99895 ] | [[ 8.999955 210.99895 ] | ||||
| [ 20.999895 241.9988 ]]]] | [ 20.999895 241.9988 ]]]] | ||||
| """ | """ | ||||
| @@ -481,10 +479,8 @@ class GlobalBatchNorm(_BatchNorm): | |||||
| >>> print(output) | >>> print(output) | ||||
| [[[[171.99915 46.999763] | [[[[171.99915 46.999763] | ||||
| [116.99941 191.99904 ]] | [116.99941 191.99904 ]] | ||||
| [[ 66.999664 250.99875 ] | [[ 66.999664 250.99875 ] | ||||
| [194.99902 102.99948 ]] | [194.99902 102.99948 ]] | ||||
| [[ 8.999955 210.99895 ] | [[ 8.999955 210.99895 ] | ||||
| [ 20.9999895 241.9988 ]]]] | [ 20.9999895 241.9988 ]]]] | ||||
| """ | """ | ||||
+ 2
- 0
mindspore/nn/metrics/fbeta.py
View File
| @@ -38,6 +38,8 @@ class Fbeta(Metric): | |||||
| >>> metric = nn.Fbeta(1) | >>> metric = nn.Fbeta(1) | ||||
| >>> metric.update(x, y) | >>> metric.update(x, y) | ||||
| >>> fbeta = metric.eval() | >>> fbeta = metric.eval() | ||||
| >>> print(fbeta) | |||||
| [0.66666667 0.66666667] | |||||
| """ | """ | ||||
| def __init__(self, beta): | def __init__(self, beta): | ||||
| super(Fbeta, self).__init__() | super(Fbeta, self).__init__() | ||||