forked from mindspore-Ecosystem/mindspore
!17878 Rectification of operator ease of use part 1
From: @dinglinhe123 Reviewed-by: @liangchenghui,@wuxuejian Signed-off-by: @liangchenghui
This commit is contained in:
commit
3fb54b93f9
|
@ -21,14 +21,15 @@ class SparseToDense(Cell):
|
|||
"""
|
||||
Converts a sparse tensor into dense.
|
||||
|
||||
Not yet supported by any backend at the moment.
|
||||
|
||||
Inputs:
|
||||
- **sparse_tensor** (:class:`mindspore.SparseTensor`): the sparse tensor to convert.
|
||||
|
||||
Outputs:
|
||||
Tensor, converted from sparse tensor.
|
||||
|
||||
Args:
|
||||
sparse_tensor (SparseTensor): the sparse tensor to convert.
|
||||
|
||||
Raises:
|
||||
TypeError: If the`sparse_tensor.indices` data type is neither int32 nor int64.
|
||||
TypeError: If the 'sparse_tensor.values' data type is not a Number or bool.
|
||||
|
|
File diff suppressed because it is too large
Load Diff
|
@ -7788,7 +7788,7 @@ class AvgPool3D(Primitive):
|
|||
>>> avg_pool3d = ops.AvgPool3D(kernel_size=2, strides=1, pad_mode="valid")
|
||||
>>> output = avg_pool3d(input)
|
||||
>>> print(output)
|
||||
[[[[[5. 6.]]]
|
||||
[[[[[ 5. 6.]]]
|
||||
[[[17. 18.]]]]]
|
||||
"""
|
||||
@prim_attr_register
|
||||
|
|
Loading…
Reference in New Issue