torch.Tensor.numpy¶
- Tensor.numpy(*, force=False) numpy.ndarray ¶
将张量作为 NumPy
ndarray
返回。如果
force
为False
(默认值),则仅当张量在 CPU 上且不需要梯度、未设置共轭位,并且是 NumPy 支持的数据类型和布局时,才会执行转换。返回的 ndarray 和张量将共享它们的存储,因此对张量的更改将反映在 ndarray 中,反之亦然。如果
force
为True
,则等效于调用t.detach().cpu().resolve_conj().resolve_neg().numpy()
。如果张量不在 CPU 上,或者设置了共轭位或负号位,则张量将不会与其返回的 ndarray 共享存储。将force
设置为True
可以作为一种有用的简写。- 参数
force (bool) – 如果为
True
,则 ndarray 可能是张量的副本,而不是始终共享内存,默认为False
。