Tensor

功能说明

片上Tensor的抽象，可指定Tensor的内存位置、数据类型、大小及排布格式作为指令的数据依赖标识。

接口原型

class Tensor(mem_type, dtype=None, size=None, format=None, is_inited=False)

参数说明

参数名	输入类型	说明
mem_type	字符串	抽象Tensor所处的内存空间的位置，如“GM”、“UB”、“L1”、“L0A”、“L0B”、“L0C”、“FB”、“BT”等。
dtype	字符串	数据类型，如BOOL、UINT1、UINT2、UINT8、UINT16、UINT32、BF16、UINT64、INT4、INT8、INT16、INT32、INT64、FP16、FP32。
size	list	Tensor的shape。
format	字符串	数据排布格式，详细可参见《Ascend C算子开发指南》的“附录 > Tensor基础知识参考 > 数据排布格式”章节。
is_inited	bool	控制Tensor类是否已就绪的开关，开启后，以该Tensor为输入的指令即可启动。

约束说明

需通过创建一个shape为[1]且is_inited=True的Tensor进行标量创建。

使用示例

from mskpp import Tensor, Core
gm_tmp= Tensor("GM", "FP16", [48, 16], format="ND")
with Core("AIV0") as aiv:  # AIV0上的相关计算逻辑
    ...
    gm_tmp.load(result, set_value=0)
with Core("AIC0") as aic:
    in_x = Tensor("GM", "FP16", [48, 16], format="ND")
    in_x.load(gm_tmp, expect_value=0) # AIC0上的相关计算逻辑 
    ...

父主题： 基础功能接口