PyTorch torch.as_tensor 函数
torch.as_tensor 是 PyTorch 中用于将数据转换为张量的函数。与 torch.tensor 不同,它尽可能共享数据内存。
函数定义
torch.as_tensor(data, dtype=None, device=None)
使用示例
实例
import torch
import numpy as np
# 从 NumPy 数组创建(共享内存)
numpy_array = np.array([1, 2, 3])
x = torch.as_tensor(numpy_array)
# 修改 NumPy 数组
numpy_array[0] = 100
print("张量:", x) # 张量也会改变
import numpy as np
# 从 NumPy 数组创建(共享内存)
numpy_array = np.array([1, 2, 3])
x = torch.as_tensor(numpy_array)
# 修改 NumPy 数组
numpy_array[0] = 100
print("张量:", x) # 张量也会改变
输出结果为:
张量: tensor([100, 2, 3])

Pytorch torch 参考手册