megengine.Parameter

class Parameter(data, dtype=None, device=None, is_const=False, no_cache=False, name=None, format='default')[源代码]

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

注解

参数上发生的操作通常返回张量而不是参数。 例如,使用参数 xx.reshape/to/sum/... 将产生一个张量。 参数和张量之间的任何操作都将以张量作为输出。

属性

T

transpose 的别名。

device

返回一个字符串,表示存储了一个 Tensor 的设备。

dmap_callback

设备映射的回调函数,参见 load

dtype

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

format

返回一个表示 Tensor 的 :ref:`memory format <format-introduction>`的字符串。

grad

此 tensor 的梯度见 autodiff

name

返回一个表示 Tensor 的名字的字符串。

ndim

返回自身 Tensor 的维数。

qparams

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

requires_grad

返回一个布尔值,表示 Tensor 是否需要梯度。

shape

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

size

返回自身 Tensor 的大小。

方法

astype(dtype)

返回一个 Tensor,其数据和元素数量与指定的 dtype 相同。

detach()

返回一个与求导当前计算图分离的新 Tensor

flatten([start_axis, end_axis])

查看 flatten

graph

item(*args)

将标准 Python numbers.Number 作为 Tensor 的值返回。

max([axis, keepdims])

See max.

mean([axis, keepdims])

See mean.

min([axis, keepdims])

See min.

numpy()

numpy.ndarray 作为self Tensor 返回。

prod([axis, keepdims])

prod

reset_zero()

1.0 版后已移除.

reshape(*args)

查看 reshape

set_value(value)

1.0 版后已移除.

sum([axis, keepdims])

sum

to(device, *[, _borrow])

Tensor 复制到指定的设备。

tolist()

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

transpose(*args)

查看 transpose

value_id

var