torch.Tensor.numpy

Tensor.numpy(*, force=False) → numpy.ndarray

将张量作为 NumPy ndarray 返回。

如果 force 为 False（默认值），仅当张量位于 CPU 上、不需要 grad、未设置共轭位且其 dtype 和布局受 NumPy 支持时，才会执行转换。返回的 ndarray 和张量将共享它们的存储，因此对张量的更改将反映在 ndarray 中，反之亦然。

如果 force 为 True，这等同于调用 t.detach().cpu().resolve_conj().resolve_neg().numpy()。如果张量不在 CPU 上，或者共轭位或负数位被设置，则张量不会与其共享存储。将 force 设置为 True 可以作为一种方便的简写。

参数

force (bool) – 如果为 True，则 ndarray 可能是张量的副本，而不是始终共享内存，默认为 False。