megengine.Parameter

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

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

注解

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

属性

T	transpose 的别名。
device	返回一个字符串，表示存储了一个 Tensor 的设备。
dmap_callback
dtype	返回一个 numpy.dtype 对象表示一个 Tensor 的数据类型。
grad
name
ndim	返回自身 Tensor 的维数。
qparams	返回一个 QParams 对象，其中包含 Tensor 的量化参数。
requires_grad
shape	返回一个 tuple 或一个 Tensor 表示张量维度。
size	返回自身 Tensor 的大小。

方法

astype(dtype)	返回一个 Tensor，其数据和元素数量与指定的 dtype 相同。
detach()	返回一个与求导当前计算图分离的新 Tensor。
flatten()	查看 flatten。
item(*args)	将标准 Python numbers.Number 作为 Tensor 的值返回。
max([axis, keepdims])	返回给定维度 axis 中输入张量的每一行的最大值。
mean([axis, keepdims])	返回给定维度 axis 中输入张量的每一行的平均值。
min([axis, keepdims])	返回给定维度 axis 中输入张量的每一行的最小值。
numpy()	将 numpy.ndarray 作为self Tensor 返回。
prod([axis, keepdims])	返回给定维度 axis 中输入张量的每一行的乘积。
reset_zero()	1.0 版后已移除.
reshape(*args)	查看 reshape。
set_value(value)	1.0 版后已移除.
sum([axis, keepdims])	返回给定维度 axis 中输入张量的每一行的总和。
to(device)	将 Tensor 复制到指定的设备。
tolist()	将张量作为（嵌套）列表返回。
transpose(*args)	查看 transpose。

上一页

megengine.Tensor.transpose

下一页

megengine.Parameter.T