mindspore.Tensor.pin_memory

Tensor.pin_memory()

Copy current Tensor to pinned memory, and return a new Tensor.

Returns

Tensor, with same elements as the input tensor.

Examples

>>> import mindspore as ms
>>> from mindspore import Tensor
>>> x = Tensor([1, 2, 3], ms.int16)
>>> out = x.pin_memory()
>>> print(out)
[1 2 3]