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 。