Tensor

函数功能

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

函数原型

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

参数说明

参数名	输入类型	说明
mem_type	字符串	抽象tensor所处的内存空间的位置，如“UB”、“L1”等。
dtype	字符串	数据类型，如BOOL、UINT1、UINT2、UINT8、UINT16、UINT32、UINT64、INT8、INT16、INT32、INT64、FP16、FP32。
size	list类型	tensor的shape。
format	字符串	数据排布格式，详细的可参见的“附录 > 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_temp.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上的相关计算逻辑
        ...

父主题： mskpp对外接口使用说明