文档
注册
评分
提单
论坛
小AI

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))
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词