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。