This commit is contained in:
lihongkang 2020-10-24 16:10:39 +08:00
parent c962ccbe07
commit 670820aaba
3 changed files with 8 additions and 3 deletions

View File

@ -334,7 +334,7 @@ class AvgPool1d(_PoolNd):
Tensor of shape :math:`(N, C_{out}, L_{out})`.
Examples:
>>> pool = nn.AvgPool1d(kernel_size=6, strides=1)
>>> pool = nn.AvgPool1d(kernel_size=6, stride=1)
>>> x = Tensor(np.random.randint(0, 10, [1, 3, 6]), mindspore.float32)
>>> output = pool(x)
>>> output.shape

View File

@ -1376,8 +1376,9 @@ class ArgMinWithValue(PrimitiveWithInfer):
- output_x (Tensor) - The minimum value of input tensor, with the same shape as index.
Examples:
>>> input_x = Tensor(np.random.rand(5))
>>> input_x = Tensor(np.random.rand(5), mindspore.float32)
>>> index, output = P.ArgMinWithValue()(input_x)
0 0.0496291
"""
@prim_attr_register

View File

@ -5740,9 +5740,13 @@ class LRN(PrimitiveWithInfer):
Tensor, with the same shape and data type as the input tensor.
Examples:
>>> x = Tensor(np.random.rand(1, 10, 4, 4)), mindspore.float32)
>>> x = Tensor(np.random.rand(1, 2, 2, 2), mindspore.float32)
>>> lrn = P.LRN()
>>> lrn(x)
[[[[0.18990143 0.59475636]
[0.6291904 0.1371534 ]]
[[0.6258911 0.4964315 ]
[0.3141494 0.43636137]]]]
"""
@prim_attr_register
def __init__(self, depth_radius=5, bias=1.0, alpha=1.0, beta=0.5, norm_region="ACROSS_CHANNELS"):