Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
magicwindyyd
mindspore
提交
87bf9a48
M
mindspore
项目概览
magicwindyyd
/
mindspore
与 Fork 源项目一致
Fork自
MindSpore / mindspore
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
mindspore
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
87bf9a48
编写于
7月 28, 2020
作者:
M
mindspore-ci-bot
提交者:
Gitee
7月 28, 2020
浏览文件
操作
浏览文件
下载
差异文件
!3585 fix sparse api doc
Merge pull request !3585 from riemann_penn/fix_sparse_api_doc
上级
b486abc0
ad2d46cd
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
24 addition
and
18 deletion
+24
-18
mindspore/common/tensor.py
mindspore/common/tensor.py
+24
-18
未找到文件。
mindspore/common/tensor.py
浏览文件 @
87bf9a48
...
...
@@ -227,6 +227,8 @@ class IndexedSlices:
IndexedSlices can only be used in `Cell`'s contruct method.
Pynative mode not supported at the moment.
Args:
indices (Tensor): A 1-D integer Tensor of shape [D0].
values (Tensor): A Tensor of any dtype of shape [D0, D1, ..., Dn].
...
...
@@ -237,16 +239,17 @@ class IndexedSlices:
IndexedSlices, composed of `indices`, `values`, `dense_shape`.
Examples:
>>> # Create a IndexedSlices.
>>> indices = Tensor([1, 2])
>>> values = Tensor([[0, 0], [1, 2]], dtype=ms.float32)
>>> dense_shape = (3, 2)
>>> indexed_slices = IndexedSlices(indices, values, dense_shape)
>>> class Net(nn.Cell):
>>> def __init__(self, dense_shape):
>>> super(Net, self).__init__()
>>> self.dense_shape = dense_shape
>>> def construct(self, indices, values):
>>> x = IndexedSlices(indices, values, self.dense_shape)
>>> return x.values(), x.indices(), x.dense_shape()
>>>
>>> # Get atrr.
>>> indices = indexed_slices.indices()
>>> values = indexed_slices.values()
>>> dense_shape = indexed_slices.dense_shape()
>>> indices = Tensor([0])
>>> values = Tensor([[1, 2]], dtype=ms.float32)
>>> Net((3, 2))(indices, values)
"""
def
__init__
(
self
,
indices
,
values
,
dense_shape
):
raise
NotImplementedError
...
...
@@ -258,6 +261,8 @@ class SparseTensor:
SparseTensor can only be used in `Cell`'s contruct method.
Pynative mode not supported at the moment.
For a tensor dense, its SparseTensor(indices, values, dense_shape) has
`dense[indices[i]] = values[i]`.
...
...
@@ -273,17 +278,18 @@ class SparseTensor:
Returns:
SparseTensor, composed of `indices`, `values`, `dense_shape`.
Examples:
>>> # Create a SparseTensor.
Examples:
>>> class Net(nn.Cell):
>>> def __init__(self, dense_shape):
>>> super(Net, self).__init__()
>>> self.dense_shape = dense_shape
>>> def construct(self, indices, values):
>>> x = SparseTensor(indices, values, self.dense_shape)
>>> return x.values(), x.indices(), x.dense_shape()
>>>
>>> indices = Tensor([[0, 1], [1, 2]])
>>> values = Tensor([1, 2], dtype=ms.float32)
>>> dense_shape = (3, 4)
>>> sparse_tensor = SparseTensor(indices, values, dense_shape)
>>>
>>> # Get atrr.
>>> indices = sparse_tensor.indices()
>>> values = sparse_tensor.values()
>>> dense_shape = sparse_tensor.dense_shape()
>>> Net((3, 4))(indices, values)
"""
def
__init__
(
self
,
indices
,
values
,
dense_shape
):
raise
NotImplementedError
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录