From b94de0ccf021eaf96905f2cb99a3baea35c24b3a Mon Sep 17 00:00:00 2001 From: brucexiaj Date: Sun, 5 Jul 2020 14:10:24 +0800 Subject: [PATCH] s --- main.py | 12 ++++++++++-- source.py | 10 +++++++++- 2 files changed, 19 insertions(+), 3 deletions(-) diff --git a/main.py b/main.py index b56cb24..d356100 100644 --- a/main.py +++ b/main.py @@ -1,4 +1,12 @@ -from source import main_func +from source import define_tensor +from mindspore import Tensor if __name__ == '__main__': - main_func() \ No newline at end of file + t1= define_tensor() + if not isinstance(t1, Tensor): + print("定义的不是张量") + if len(t1.shape) != 3: + print("定义的张量不是三阶张量") + if t1.dtype != mindspore.int32 + print("定义的张量不是int32类型") + print("定义张量成功") \ No newline at end of file diff --git a/source.py b/source.py index 5f28270..f2c8f74 100644 --- a/source.py +++ b/source.py @@ -1 +1,9 @@ - \ No newline at end of file +import mindspore +from mindspore import Tensor +def define_tensor(): + # 定义一个int32类型的三阶张量,内容不限 + # 将这个张量作为函数的返回值 + ########## Begin ########## + tensor = Tensor([[[1, 2], [3, 4]], [[5, 6], [7, 8]]], mindspore.int32) # 定义一个张量 + return tensor + ########## End########## \ No newline at end of file