和PyTorch forum 本质上, torch.Tensor.expand() 是您正在寻找的功能,可以按如下方式使用: x = torch.rand([39, 1, 20, 256, 256]) y = x.expand(39, 3, 20, 256, ...
確定! 回上一頁