更新时间:2021-03-18 GMT+08:00
分享

reshape

功能说明

设置或者改变Tensor shape。

函数原型

reshape(new_shape)

参数说明

表1 参数说明

参数名称

输入/输出

含义

new_shape

输入

指定新的Tensor对象的形状,支持List(int),tuple(int)类型。

须知:

当前版本由整数立即数组成的list或者tuple。

注意事项

  • 新的shape的总大小和原来定义的shape的大小必须一致。
  • 新tensor和原来的tensor内存指向同一块。新tensor的值修改后,原来的tensor值也会改变。

返回值

新的tensor。

调用示例

from te import tik
tik_instance = tik.Tik()
data_A = tik_instance.Tensor("float16", (128,), name="data_A", scope=tik.scope_gm)
data_A.reshape((64,2))
分享:

    相关文档

    相关产品

关闭导读