torch.from_numpy

这篇文章介绍了PyTorch库中的torch.from_numpy函数,它将NumPy数组无缝转换为Tensor,共享底层内存以提高处理大型数据集的效率。
摘要由CSDN通过智能技术生成

torch.from_numpy 是 PyTorch 中的一个函数,用于将 NumPy 数组转换为 PyTorch 的 Tensor。这个函数主要用于共享底层数据内存,而不进行数据的复制。

具体用法如下:

import torch
import numpy as np

# 创建一个 NumPy 数组
numpy_array = np.array([1, 2, 3])

# 使用 torch.from_numpy 将 NumPy 数组转换为 PyTorch Tensor
torch_tensor = torch.from_numpy(numpy_array)

在上述示例中,torch_tensor 将是一个 PyTorch Tensor,与 numpy_array 共享底层的数据内存。

需要注意的是,torch.from_numpy 的输入应该是一个 NumPy 数组。如果输入是一个 Python 列表,需要先将列表转换为 NumPy 数组,然后再使用 torch.from_numpy

python_list = [1, 2, 3]

# 将 Python 列表转换为 NumPy 数组
numpy_array = np.array(python_list)

# 使用 torch.from_numpy 将 NumPy 数组转换为 PyTorch Tensor
torch_tensor = torch.from_numpy(numpy_array)

这个函数在处理大型数据集时很有用,因为它避免了不必要的数据复制,提高了效率。

  • 7
    点赞
  • 13
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值