废话不多说直接上图如图有a, b, c 三个3x3的Tensor, 如果我想把这三个tensor的最后一个维度的元素相叠加,形成一个新的tensor输入d=torch.stack( (a,b ...
確定! 回上一頁