Pytorch 扩展tensor维度

1、扩展1维

input = torch.randn(256, 192, 192)
input = torch.unsqueeze(input, 0)

2、扩展多维

input = torch

你可能感兴趣的:(pytorch实践,pytorch)