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]