PyTorch solution for TimeDistributedDense

How do I realize something similar to keras TimeDistributedDense ( in pytorch?

Because of pytorch’s dynamic graph, you don’t need TimeDistributedDense like in Keras. You can just use Linear. LSTM networks become very straight forward. See the tutorial for pos tagging with LSTM:

1 Like
# 24 fc timedistributed
num = 24
fc = nn.ModuleList([nn.Linear(8, 1) for i in range(num)])

# forward pass 
x = np.zeros(64, 24, 8)
for i in range(x.shape[1]):
 outs.append(fc[i](x[:, i, :].unsqueeze(1))), axis=1)

Hi, could you further explain more about it?