Screenshot (685)1920×1080 236 KB

why it’s not printing the shape of anything else?
``` def onehot_encode(label, device, n_class=n_class):
eye = torch.eye(n_class, device=device)
return eye[label].view(-1, n_class, 1, 1)
def concat_image_label(image, label, device, n_class=n_class):
B, C, H, W = image.shape
oh_label = onehot_encode(label, device=device)
oh_label = oh_label.expand(B, n_class, H, W)
return torch.cat((image, oh_label), dim=1)
def concat_noise_label(noise, label, device):
oh_label = onehot_encode(label, device=device)
return torch.cat((noise, oh_label), dim=1) ```
as well is this causing any issue???