import torch
a = torch.Tensor(4, 1 , 999)
a = a[:, :, 0]
print(a.shape)
将torch.size([a, b, c])变成torch.size([a, b])
最新推荐文章于 2024-07-12 16:21:48 发布
import torch
a = torch.Tensor(4, 1 , 999)
a = a[:, :, 0]
print(a.shape)