我有一个大小为: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)
关于python - 如何根据模运算在 PyTorch 张量的末尾添加零?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/64354309/