logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

如何给一个二维的Tensor添加一列或者一行呢

PyTorch的使用学习问题:如何给一个二维的Tensor添加一列或者一行呢?解答:用torch.unsqueeze和torch.cat例:import torchn = torch.randn([6,4])p = torch.ones(n.shape[0])p = p.unsqueeze(1)print(n)print(p)n = torch.cat((n,p),1)print(n)因为不同维度

#python
暂无文章信息