现在位置: 首页 > PyTorch 教程 > 正文

PyTorch torch.as_tensor 函数


Pytorch torch 参考手册 Pytorch torch 参考手册

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)  # 张量也会改变

输出结果为:

张量: tensor([100,   2,   3])

Pytorch torch 参考手册 Pytorch torch 参考手册