The dim of the data

[RuifMaxx]

When load the data from the dataloader, the dim is transposed:
data = Variable(data.squeeze().transpose(0, 1)).to(device)
So in model.py line 82

 for t in range(x.size(0)):

	phi_x_t = self.phi_x(x[t])

the size of phi_x_t is ( batch_size, h_dim ), is that correct ?