将两个Tensor组合在一起的规范/推荐方法是什么?例如,如果我们有两个2DTensor,我希望对应的部分沿着dim=1彼此相邻:
nukf8bse1#
这里有一个方法:
import torch a = torch.tensor([[1, 2],[3, 4]]) b = 10 * a c = torch.zeros((a.shape[0], 2*a.shape[1]), dtype=a.dtype) c[:, 0::2] = a c[:, 1::2] = b print("a:\n", a) print("b:\n", b) print("c:\n", c)
结果是:
a: tensor([[1, 2], [3, 4]]) b: tensor([[10, 20], [30, 40]]) c: tensor([[ 1, 10, 2, 20], [ 3, 30, 4, 40]])
1条答案
按热度按时间nukf8bse1#
这里有一个方法:
结果是: