pytoch RuntimeError: Dimension out of range (expected to be in range of [-1, 0], but got 1

Question

我正在尝试在演员和评论家中使用 LSTM 训练演员评论家模型。我对这一切都是陌生的，无法理解为什么 "RuntimeError: Dimension out of range (expected to be in range of [-1, 0], but got 1)" 即将到来。

我正从 actor 向前传播并收到错误

下面是我的代码和错误 message.I 我正在使用 pytorch 版本 0.4.1

谁能帮忙检查一下这段代码有什么问题。

import os
import time
import random
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
from sklearn.preprocessing import StandardScaler
import torch
import torch.nn as nn
import torch.nn.functional as F
from random import random as rndm
from torch.autograd import Variable
from collections import deque
   
torch.set_default_tensor_type('torch.DoubleTensor')


class Actor(nn.Module):
  
  def __init__(self, state_dim, action_dim, max_action):
    super(Actor, self).__init__()
    self.lstm = nn.LSTMCell(state_dim, 256)
    self.layer_1 = nn.Linear(256, 400)
    self.layer_2 = nn.Linear(400, 300)
    self.layer_3 = nn.Linear(300, action_dim)
    self.hx = torch.zeros(1,256)
    self.cx = torch.zeros(1,256)
    self.max_action = max_action

  def forward(self, x):
    self.hx, self.cx = self.lstm(x, (self.hx, self.cx))
    x = F.relu(self.layer_1(self.hx))
    x = F.relu(self.layer_2(x))
    x = self.max_action * torch.tanh(self.layer_3(x))
    return x

state_dim = 3
action_dim = 3
max_action = 1

policy = Actor(state_dim, action_dim, max_action)

s = torch.tensor([20,20,100])
next_action = policy(s)

错误信息是：

next_action = policy(s)
Traceback (most recent call last):

  File "<ipython-input-20-de717f0ad3d2>", line 1, in <module>
    next_action = policy(s)

  File "C:\Users\granthjain\anaconda3\lib\site-packages\torch\nn\modules\module.py", line 477, in __call__
    result = self.forward(*input, **kwargs)

  File "<ipython-input-4-aed4daf511cb>", line 14, in forward
    self.hx, self.cx = self.lstm(x, (self.hx, self.cx))

  File "C:\Users\granthjain\anaconda3\lib\site-packages\torch\nn\modules\module.py", line 477, in __call__
    result = self.forward(*input, **kwargs)

  File "C:\Users\granthjain\anaconda3\lib\site-packages\torch\nn\modules\rnn.py", line 704, in forward
    self.check_forward_input(input)

  File "C:\Users\granthjain\anaconda3\lib\site-packages\torch\nn\modules\rnn.py", line 523, in check_forward_input
    if input.size(1) != self.input_size:

RuntimeError: Dimension out of range (expected to be in range of [-1, 0], but got 1)

我正在使用 pytorch 版本 0.4.1

谁能帮忙检查一下这段代码有什么问题。

Answer 1

知道了。

lstm 层的输入有不同的形状。 https://pytorch.org/docs/master/generated/torch.nn.LSTMCell.html

pytoch RuntimeError: Dimension out of range (expected to be in range of [-1, 0], but got 1

pytoch RuntimeError: Dimension out of range (expected to be in range of [-1, 0], but got 1

python

artificial-intelligence

reinforcement-learning

pytorch