megengine.Tensor¶
- class Tensor(data, dtype=None, device=None, is_const=False, no_cache=False, name=None)[源代码]¶
一个张量对象表示固定大小元素的多维同构数组。
- 参数
属性
transpose
的别名。返回一个字符串,表示当前
Tensor
所存储在的设备。返回一个
numpy.dtype
对象,表示当前Tensor
的数据类型。返回自身
Tensor
的维数。返回当前
Tensor
的大小。方法
astype
(dtype)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])返回当前张量在给定轴上所有元素求乘积后的结果。
1.0 版后已移除.
reshape
(*args)参考
reshape
.set_value
(value)1.0 版后已移除.
sum
([axis, keepdims])返回当前张量在给定轴上所有元素求和后的结果。
to
(device)将当前
Tensor
拷贝到指定设备。tolist
()返回将当前张量转换成(嵌套)列表后的结果。
transpose
(*args)参考
transpose
.