例如,我有一个向量 [1, 2, 3]
我想得到一个矩阵,像这样
[[1, 2, 3],[1, 2, 3],[1, 2, 3],[1, 2, 3]]
如何高效地做到这一点?
回答:
你应该查看 .repeat()
方法:
In [1]: torch.Tensor([1, 2, 3]).repeat(4, 1)Out[1]: tensor([[1., 2., 3.], [1., 2., 3.], [1., 2., 3.], [1., 2., 3.]])