摘要:转载Pytorch:tensor.mean()和tensor.sum()_三世的博客-CSDN博客 import torch x = torch.arange(24).view(2, 3, 4).float() y = x.mean(0) print("x.shape:", x.shape) pri
阅读全文
posted @ 2023-11-20 15:50
posted @ 2023-11-20 15:50