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	返回一个 tuple 或 Tensor, 表示当前张量的维度信息。
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.

上一页

megengine

下一页

megengine.Tensor.T