无涯教程网

Python TensorDataset 给出错误int对象不可调用，但所有输入都是 numpy 数组

发布于05月20日

我正在将TensorFlow项目转换为PyTorch，但我遇到了一个奇怪的错误.当我试图创造一个像这样的TensorDataset

train_data = TensorDataset(
    self.x_train[idx_train], 
    self.covariates_train[idx_train],
    self.y_train[idx_train]
)

torch 掷出TypeError: 'int' object is not callable.相应的TensorFlow代码可以正常工作.我还判断了输入，它们的形状都是正确的.具体地说，每个参数的形状如下:

Argument 1:  (200, 1, 40, 26) (180,) (180, 1, 40, 26)
Argument 2:  (200, 2) (180, 2)
Argument 3:  (200, 51) (180, 51)

这些形状与对应的TensorFlow代码在代码中的类似点匹配(即，当我要将这些相同的输入输入到tf.data.Dataset.from_tensor_slices中时).

推荐答案

使用torch.utils.data.TensorDataset时，应将输入转换为torch.Tensor:

import torch
import numpy as np

x_train = np.random.uniform(size=(200, 1, 40, 26))
y_train = np.random.uniform(size=(200, 51))

dataset = torch.utils.data.TensorDataset(
    torch.from_numpy(x_train),
    torch.from_numpy(y_train)
)

对于除了tf.Tensor之外还支持不同数据类型的tf.data.Dataset.from_tensor_slices，这不是必需的.

Python相关问答推荐

Django文件上传不起作用：文件未出现在媒体目录或数据库中

KNN分类器中的GridSearchCV

有没有方法可以修复删除了换码字符的无效的SON记录？

返回nxon矩阵的diag元素，而不使用for循环

如何计算两极打印机中 * 所有列 * 的出现次数？

Django管理面板显示字段最大长度而不是字段名称

我从带有langchain的mongoDB中的vector serch获得一个空数组

连接两个具有不同标题的收件箱

ModuleNotFound错误：没有名为Crypto Windows 11、Python 3.11.6的模块

Pandas 有条件轮班操作

大小为M的第N位_计数(或人口计数)的公式

如何在类和classy-fastapi -fastapi- followup中使用FastAPI创建路由

在Wayland上使用setCellWidget时，try 编辑QTable Widget中的单元格时，PyQt 6崩溃

改进大型数据集的框架性能

Python逻辑操作作为Pandas中的条件

Python导入某些库时非法指令(核心转储)(beautifulsoup4."" yfinance)

如何更改groupby作用域以找到满足掩码条件的第一个值？

Flash只从html表单中获取一个值

将标签移动到matplotlib饼图中楔形块的开始处

Pandas：填充行并删除重复项，但保留不同的值

实用课程推荐

相关教程推荐