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 上，或者設定了共軛位或負位，則張量不會與其返回的 ndarray 共享儲存。將 force 設定為 True 可以是一種方便的快捷方式。

引數

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