mindspore.Tensor.pin_memory

查看源文件
mindspore.Tensor.pin_memory()

将当前Tensor的数据拷贝到锁页内存中,并返回一个新的Tensor。

返回:

Tensor,含有与输入相同的元素。

样例:

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