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

callback for device mapping, see load.

dtype

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

format

Returns a string represents the memory format of a Tensor.

grad

gradient of this tensor, see autodiff.

name

Returns a string represents the name of a Tensor.

ndim

返回自身 Tensor 的维数。

qparams

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

requires_grad

Returns a bool indicates whether the Tensor requires gradient.

shape

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

size

返回自身 Tensor 的大小。

方法

astype(dtype)

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

detach()

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

flatten()

查看 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])

See prod.

reset_zero()

1.0 版后已移除.

reshape(*args)

查看 reshape

set_value(value)

1.0 版后已移除.

sum([axis, keepdims])

See sum.

to(device, *[, _borrow])

Tensor 复制到指定的设备。

tolist()

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

transpose(*args)

查看 transpose

var