昇腾社区首页
中文
注册
开发者
下载

set_shape

功能说明

设置张量形状。

函数原型

1
set_shape(shape: List[int]) -> TensorHolder

参数说明

参数名

输入/输出

描述

shape

输入

形状维度列表。

返回值说明

(TensorHolder)返回自身以支持链式调用。

约束说明

  • 如果shape不是整数列表,抛出TypeError。
  • 如果所有形状维度不是整数,抛出TypeError。
  • 如果操作失败,抛出RuntimeError。