报错“ValueError: only one element tensors can be converted to Python scalar

用了append报错,将apend换成extend

        loss, I_ZY_bound, I_ZX_bound, I_ZY_bound_2 = model.batch_loss(x_batch, y_batch)
        train_loss.append(loss.item())
        logits, logits2 = model.forward(x_batch)

猜你喜欢

转载自blog.csdn.net/Viviane_2022/article/details/128692361