实际上torch.FloatTensor()的输入最好是列表等,而不是一个数字。
参考:
https://blog.csdn.net/qq_27261889/article/details/103186904