megengine.Tensor.shape

property Tensor.shape: Union[tuple, megengine.tensor.Tensor]

返回一个用于表示张量维度的 tupleTensor .

注解

张量的形状通常用 tuple 进行表示,但如果某张量在 tracing 的时候被作为占位符,其形状也可以用 Tensor 来表示。查看 trace 了解更多细节。

Shape 属性通常用于获取张量的当前形状,但也可以通过向其赋值一个表示形状的 tuple 来对原张量进行形状的原地改变。就像 reshape 一样,新形状的某个维度可以为 -1,在这种情况下,其对应值可以从张量的大小和其余维度推断出来。

返回类型

Union[tuple, Tensor]