I would like to look at my decoder weight matrix ( as a heatmap). I am a bit confused about the shapes, shouls I user the 5th matrix? ( the size can be seen in the model.parameters().
Model(
  (encoder): Encoder(
    (model): ModuleList(
      (0): Linear(in_features=6000, out_features=100, bias=True)
      (1): BatchNorm1d(100, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
      (2): ReLU()
      (3): Dropout(p=0.2)
    )
  )
  (decoder): Decoder(
    (model): ModuleList(
      (0): Linear(in_features=100, out_features=6000, bias=True)
      (1): BatchNorm1d(6000, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
      (2): ReLU()
      (3): Dropout(p=0.2)
    )
  )
)

model parameters size:

torch.Size([100, 6000])
torch.Size([100])
torch.Size([100])
torch.Size([100])
torch.Size([6000, 100])
torch.Size([6000])
torch.Size([6000])
torch.Size([6000])