torch用复数训练

这个垃圾torch竟然不能用复数训练。请注意，torch是有复数形式的，在dataset里面是可以用这一数据形式的，但是！！！在模型中是不可以的，在gpu训练时只能用float类型进行训练，所以，我原来【256,999】的数据形状就会变成【256,999,2】这个后面的2就是实部和虚部！！！所以我打算直接把这个当做2通道进行处理，也就是说，把数据形状变成【2,256,999】这样去处理应该就OK。这个最好是在dataset里面进行处理。新建一个数组，把实部和虚部分别放入。然后你还得把数据转成float32的。x=x.astype(dtype=np.float32)。现在就OK了。然后可以去调模型里面的数组形状了。垃圾torch还要自己算输入参数数量，真是反人类。