tensorr

Tancenter5 小时前
pytorch·tensorr
sequeeze()和unsequeeze()这是最常见的用法，作用是‌去掉张量中维度为 1 的轴‌。‌import torch x = torch.randn(1, 3, 1, 4) print(x.shape) # torch.Size([1, 3, 1, 4]) print(x.squeeze().shape) # torch.Size([3, 4])
我是有底线的