Any PyTorch function can work as Keras' Timedistributed?

2d CNN accepts 4d inputs only so you can pass the 5d tensor ( batch , timesteps , channles ,height ,width) as 4d tensor

view(batch*timesteps,c,h,w)
1 Like