with open('your_file_path.txt', 'r') as file: data = file.read()将数据读为tensor类型
时间: 2024-02-04 18:03:50 浏览: 161
从txt中读入数据
您可以使用以下代码将读入的数据转换为PyTorch张量:
```
import torch
with open('your_file_path.txt', 'r') as file:
data = file.read()
data = data.strip().split() # split the data into a list of strings
tensor_data = torch.tensor([int(d) for d in data]) # convert the list to a PyTorch tensor
```
请注意,此代码假设文件中的数据已经是数字,每个数字之间使用空格分隔。如果数据以其他格式存在,请根据需要修改代码。
阅读全文