Pytorch增加维度

版权声明:本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明出处! https://blog.csdn.net/huplion/article/details/81029369

增加维度的例子

sig = torch.from_numpy(sig).float().unsqueeze(0)

猜你喜欢

转载自blog.csdn.net/huplion/article/details/81029369