# Codeblock 6 Outputimages                  : torch.Size([2, 1, 28, 28])   #(1)timesteps               : torch.Size([2]), tensor([468, 304], device='cuda:0')  #(2)after first conv        : torch.Size([2, 64, 28, 28])  #(3)time_embed              : torch.Size([2, 32])          #(4)time_embed after linear : torch.Size([2, 64])time_embed expanded     : torch.Size([2, 64, 1, 1])    #(5)after summation         : torch.Size([2, 64, 28, 28])  #(6)after second conv       : torch.Size([2, 64, 28, 28])  #(7)