I understand that that output “torch.Size([1024])” has an error, It is needed to become "torch.Size([128,8])
it appears be flatten.
I don’t use  the gym , PLE ,etc.
I refer to tutorial(Reinforcement Learning (DQN) Tutorial — PyTorch Tutorials 1.7.1 documentation 2),
inputs is equal to tutorial’s state.
inputs =  [A,B,C,D,E,F,G,H]
inputs = np.array(inputs)
inputs = torch.from_numpy(inputs).type(torch.FloatTensor).to(device)

I can’t run my progam yet.
Please Help me.