reshape
功能说明
改变Tensor shape。
函数原型
reshape(new_shape)
参数说明
参数名称 |
输入/输出 |
含义 |
---|---|---|
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管理