torch.from_numpy
2024-01-08 19:00:18
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)
这个函数在处理大型数据集时很有用,因为它避免了不必要的数据复制,提高了效率。
文章来源:https://blog.csdn.net/Recursions/article/details/135462235
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。 如若内容造成侵权/违法违规/事实不符,请联系我的编程经验分享网邮箱:veading@qq.com进行投诉反馈,一经查实,立即删除!
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。 如若内容造成侵权/违法违规/事实不符,请联系我的编程经验分享网邮箱:veading@qq.com进行投诉反馈,一经查实,立即删除!