2023年4月28日
y_hat[[0, 1], y]
摘要： y = torch.tensor([0, 2]) y_hat = torch.tensor([[0.1, 0.3, 0.6], [0.3, 0.2, 0.5]]) y_hat[[0, 1], y] 这段代码使用了两个 PyTorch 张量 y 和 y_hat，并通过 y_hat[[0, 1], y] 阅读全文
posted @ 2023-04-28 22:11 要多读书的陈小派 阅读(302) 评论(0) 推荐(0)