megengine.Tensor

class Tensor(data, dtype=None, device=None, is_const=False, no_cache=False, name=None)[源代码]

一个张量对象表示固定大小元素的多维同构数组。

参数
  • data (Tensor, ndarray, list or python number.) – 返回张量的值。

  • dtype (Optional[dtype]) – 张量的类型。如果没有特别指定,则使用数据的类型。

  • device (Optional[str]) – 张量的所在设备。如果没有特别指定则为 get_default_device

  • is_const (bool) – 是否在 tracing 的模式下作为一个不可变张量。

  • no_cache (bool) – 是否为了内存共享缓存张量。

  • name (Optional[str]) – 张量的名字可用来简化存储的模型上的图操作。

属性

T

transpose 的别名。

c_name

device

返回一个字符串,表示当前 Tensor 所存储在的设备。

dmap_callback

dtype

返回一个 numpy.dtype 对象,表示当前 Tensor 的数据类型。

grad

name

ndim

返回自身 Tensor 的维数。

qparams

返回一个 QParams 对象,其中包含当前 Tensor 的量化参数。

requires_grad

shape

返回一个 tupleTensor, 表示当前张量的维度信息。

size

返回当前 Tensor 的大小。

方法

astype(dtype)

返回一个含有完全相同的数据和元素数目,但是数据类型被 dtype 指定的新 Tensor.

detach()

返回一个与当前图解绑后的新 Tensor .

flatten()

参考 flatten.

item(*args)

以标准 Python numbers.Number 类型返回当前 Tensor 元素的值。

max([axis, keepdims])

返回当前张量在给定轴上所有元素求最大值后的结果。

mean([axis, keepdims])

返回当前张量在给定轴上所有元素求平均值后的结果。

min([axis, keepdims])

返回当前张量在给定轴上所有元素求最小值后的结果。

numpy()

返回一个将当前 Tensor 转换为 numpy.ndarray 后的结果。

prod([axis, keepdims])

返回当前张量在给定轴上所有元素求乘积后的结果。

reset_zero()

1.0 版后已移除.

reshape(*args)

参考 reshape.

set_value(value)

1.0 版后已移除.

sum([axis, keepdims])

返回当前张量在给定轴上所有元素求和后的结果。

to(device)

将当前 Tensor 拷贝到指定设备。

tolist()

返回将当前张量转换成(嵌套)列表后的结果。

transpose(*args)

参考 transpose.