张量(Tensor)

Tensor

一个固定维度的多维同构数组

tensor

megengine.tensor.Tensor 的别名

Parameter

一类表示模块参数的张量。

张量的类型与形状

Tensor.dtype

返回一个 numpy.dtype 的数据类型,用于表示 Tensor

Tensor.astype

返回一个 Tensor 类型,它拥有与原数据相同大小的尺寸和形状,但是变为指定的 :attr:`~.Tensor.dtype`类型

Tensor.shape

返回一个 tuple 或者 Tensor 表示张量的维度。

Tensor.reshape

可查看 reshape

Tensor.ndim

返回自身 Tensor 的维数。

Tensor.size

返回张量 Tensor 自身的大小。

Tensor.flatten

可查看 flatten

Tensor.transpose

可查看 transpose

Tensor.T

transpose 的别名。

转为其它数据类型

Tensor.numpy

将张量 Tensor 的值作为 numpy.ndarray 返回。

Tensor.tolist

将张量值作为 (可嵌套的)列表返回。

Tensor.item

将张量 Tensor 作为标准 Python numbers.Number 返回。

归约计算(Reduction)

Tensor.sum

返回给定维度 axis 中输入张量的每一行的总和。

Tensor.mean

返回给定维度 axis 中输入张量的每一行的平均值。

Tensor.prod

返回给定维度 axis 中输入张量的每一行的乘积。

Tensor.min

返回给定维度 axis 中输入张量的每一行的最小值。

Tensor.max

返回给定维度 axis 中输入张量的每一行的最大值。

求导时视作常量

Tensor.detach

返回一个与当前计算图分离的新张量。

查询/改变所在设备

Tensor.device

返回一个字符串表示的当前张量所在的设备。

Tensor.to

将当前张量拷贝到指定设备。

*对张量进行计算

参见