2维array或DataFrame转换为tensor用于pytorch训练

Question

使用数据格式为（X，Y）X为样本数量，Y为数据长度，想转换为（X，1，Y）的格式用于训练，请问怎么转换。

Answer 1

可以使用PyTorch中的unsqueeze()函数将数据从(X,Y)转换为(X,1,Y)的格式。