我有一个张量大小： torch.Size([1, 305760])
由于305760不能被400整除，因此我想将大小更改为：[1, 306000]用填充剩余的空间0。如何使用PyTorch做到这一点？
a = torch.randn(1,305760)
N = a.shape[1]
M = 400
b = torch.zeros([a.shape[0], (N // M + 1 ) * 400 - N])
a = torch.cat((a, b), 1)
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句