如何将熊猫数据框转换为张量并重塑它?

乘风 pytorch 196

原文标题How can I convert pandas dataframe to tensor and reshape it?

我想将一些数据框更改为张量以使其在 rnn 模型上工作。数据的原始形状是这样的。

input : X_train.shape
output : (2101, 32)

并且预期的形状是

output : (2101, 4, 8)

我试过了

X_train.to_numpy.reshape(-1, 4, 8)

但它没有用。

请帮助

原文链接:https://stackoverflow.com//questions/71417959/how-can-i-convert-pandas-dataframe-to-tensor-and-reshape-it

回复

我来回复
  • iacob的头像
    iacob 评论

    您正在寻找numpy方法:

    X_train.reshape(-1, 4, 8).numpy()
    
    2年前 0条评论