简介
该用户还未填写简介
擅长的技术栈
未填写擅长的技术栈
可提供的服务
暂无可提供的服务
如何给一个二维的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)因为不同维度
暂无文章信息