torch.Tensor.numpy

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

將 tensor 返回為 NumPy ndarray。

如果 force 為 False (預設值)，則僅當 tensor 在 CPU 上、不計算梯度、沒有設定共軛位、並且是 NumPy 支援的 dtype 和 layout 時，才執行轉換。返回的 ndarray 和 tensor 將共享它們的 storage，因此對 tensor 的更改將反映在 ndarray 中，反之亦然。

如果 force 為 True，則相當於呼叫 t.detach().cpu().resolve_conj().resolve_neg().numpy()。如果 tensor 不在 CPU 上，或設定了共軛位或負號位，tensor 將不會與返回的 ndarray 共享其 storage。將 force 設定為 True 可以作為一種有用的簡寫方式。

引數

force (bool) – 如果為 True，則 ndarray 可能是 tensor 的副本而不是始終共享記憶體，預設為 False。