Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
magicwindyyd
mindspore
提交
8c5db0f9
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看板
提交
8c5db0f9
编写于
8月 06, 2020
作者:
B
buxue
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add attr 'shape' and 'dtype' and interface 'asnumpy' for Tensor
上级
b6fe2577
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
32 addition
and
14 deletion
+32
-14
mindspore/ccsrc/utils/tensor_py.cc
mindspore/ccsrc/utils/tensor_py.cc
+2
-2
mindspore/common/tensor.py
mindspore/common/tensor.py
+30
-12
未找到文件。
mindspore/ccsrc/utils/tensor_py.cc
浏览文件 @
8c5db0f9
...
...
@@ -268,7 +268,7 @@ REGISTER_PYBIND_DEFINE(Tensor, ([](const py::module *m) {
}),
py
::
arg
(
"input"
),
py
::
arg
(
"dtype"
)
=
nullptr
)
.
def_property
(
"init_flag"
,
&
Tensor
::
is_init
,
&
Tensor
::
set_init_flag
)
.
def_property_readonly
(
"dtype"
,
&
Tensor
::
Dtype
,
R"mydelimiter(
.
def_property_readonly
(
"
_
dtype"
,
&
Tensor
::
Dtype
,
R"mydelimiter(
Get the tensor's data type.
Returns:
...
...
@@ -279,7 +279,7 @@ REGISTER_PYBIND_DEFINE(Tensor, ([](const py::module *m) {
>>> data.dtype
Int32
)mydelimiter"
)
.
def_property_readonly
(
"shape"
,
TensorPy
::
GetPyTupleShape
,
R"mydelimiter(
.
def_property_readonly
(
"
_
shape"
,
TensorPy
::
GetPyTupleShape
,
R"mydelimiter(
Get the tensor's shape.
Returns:
...
...
mindspore/common/tensor.py
浏览文件 @
8c5db0f9
...
...
@@ -208,13 +208,41 @@ class Tensor(Tensor_):
return
"Unknown Tensor type!"
return
str
(
self
.
asnumpy
())
@
property
def
shape
(
self
):
"""The shape of tensor."""
return
self
.
_shape
@
property
def
dtype
(
self
):
"""The dtype of tensor."""
return
self
.
_dtype
@
property
def
virtual_flag
(
self
):
"""Mark tensor is virtual."""
return
self
.
_virtual_flag
@
virtual_flag
.
setter
def
virtual_flag
(
self
,
value
):
"""The setter of virtual_flag."""
if
not
isinstance
(
value
,
bool
):
raise
TypeError
(
"virtual_flag must be bool."
)
self
.
_virtual_flag
=
value
def
asnumpy
(
self
):
"""Convert tensor to numpy array."""
return
Tensor_
.
asnumpy
(
self
)
def
all
(
self
,
axis
=
(),
keep_dims
=
False
):
"""
Check all array elements along a given axis evaluate to True.
Args:
axis (Union[None, int, tuple(int)): Dimensions of reduction.
Default: (), reduce all dimensions.
keep_dims (bool): Whether to keep the reduced dimensions.
Default : False, don't keep these reduced dimensions.
Returns:
Tensor, has the same data type as x.
...
...
@@ -228,7 +256,9 @@ class Tensor(Tensor_):
Args:
axis (Union[None, int, tuple(int)): Dimensions of reduction.
Default: (), reduce all dimensions.
keep_dims (bool): Whether to keep the reduced dimensions.
Default : False, don't keep these reduced dimensions.
Returns:
Tensor, has the same data type as x.
...
...
@@ -236,18 +266,6 @@ class Tensor(Tensor_):
return
tensor_operator_registry
.
get
(
'any'
)(
keep_dims
)(
self
,
axis
)
@
property
def
virtual_flag
(
self
):
"""Mark tensor is virtual."""
return
self
.
_virtual_flag
@
virtual_flag
.
setter
def
virtual_flag
(
self
,
value
):
"""The setter of virtual_flag."""
if
not
isinstance
(
value
,
bool
):
raise
TypeError
(
"virtual_flag must be bool."
)
self
.
_virtual_flag
=
value
class
IndexedSlices
:
"""
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录