megengine.Parameter

class Parameter(data, dtype=None, device=None, is_const=False, no_cache=False, name=None, format='default')

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

注解

参数上发生的操作通常返回张量而不是参数。 例如，使用参数 x，x.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

上一页

megengine.Tensor.var

下一页

megengine.Parameter.T