ptrblck:

print(tensor.shape)


print(loss.shape)
torch.Size([])
tensor([[[0, 0, 0,  …, 0, 0, 0],
[0, 0, 0,  …, 0, 0, 0],
[0, 0, 0,  …, 0, 0, 0],
…,
[0, 0, 0,  …, 0, 0, 0],
[0, 0, 0,  …, 0, 0, 0],
[0, 0, 0,  …, 0, 0, 0]],
    [[0, 0, 0,  ..., 0, 0, 0],
     [0, 0, 0,  ..., 0, 0, 0],
     [0, 0, 0,  ..., 0, 0, 0],
     ...,
     [0, 0, 0,  ..., 0, 0, 0],
     [0, 0, 0,  ..., 0, 0, 0],
     [0, 0, 0,  ..., 0, 0, 0]],

##############################
print(predicted.shape)
torch.Size([10, 240, 320])
###########################
print(t_image.shape)
torch.Size([10, 1, 240, 320])
############################