reshape

功能说明

改变Tensor shape。

函数原型

reshape(new_shape)

参数说明

表1 参数说明
参数名称	输入/输出	含义
new_shape	输入	指定新的Tensor对象的形状，支持List、Tuple类型，List、Tuple中的元素可以为立即数（int）、Scalar（int、uint）或Expr（int、uint）。

支持的型号

Atlas 200/300/500 推理产品

Atlas 训练系列产品

Atlas推理系列产品（Ascend 310P处理器）AI Core

Atlas推理系列产品（Ascend 310P处理器）Vector Core

Atlas A2训练系列产品/Atlas 800I A2推理产品

Atlas 200/500 A2推理产品

注意事项

若一个Tensor是由别的Tensor获取部分数据而得来的，那么暂不支持该Tensor的reshape操作。
新的shape的总大小和原来定义的shape的总大小必须一致。
新Tensor和原来的Tensor内存指向同一块。新Tensor的值修改后，原来的Tensor值也会改变。

返回值

新的Tensor。

调用示例

from tbe import tik
tik_instance = tik.Tik()
data_A = tik_instance.Tensor("float16", (128,), name="data_A", scope=tik.scope_gm)
data_a_reshape = data_A.reshape((64,2))

父主题： Tensor管理