- Tensor.numpy(*, force=False) numpy.ndarray ¶
Returns the tensor as a NumPy
False(the default), the conversion is performed only if the tensor is on the CPU, does not require grad, does not have its conjugate bit set, and is a dtype and layout that NumPy supports. The returned ndarray and the tensor will share their storage, so changes to the tensor will be reflected in the ndarray and vice versa.
Truethis is equivalent to calling
t.detach().cpu().resolve_conj().resolve_neg().numpy(). If the tensor isn’t on the CPU or the conjugate or negative bit is set, the tensor won’t share its storage with the returned ndarray. Setting
Truecan be a useful shorthand.
force (bool) – if
True, the ndarray may be a copy of the tensor instead of always sharing memory, defaults to