在pytorch中重复3d张量的行

import torch

# 创建一个3D张量
tensor = torch.tensor([[[1, 2, 3],
                        [4, 5, 6],
                        [7, 8, 9]]])

# 使用repeat()函数重复行
repeated_tensor = tensor.repeat(1, 3, 1)

print(repeated_tensor)

tensor([[[1, 2, 3],
         [1, 2, 3],
         [1, 2, 3],
         [4, 5, 6],
         [4, 5, 6],
         [4, 5, 6],
         [7, 8, 9],
         [7, 8, 9],
         [7, 8, 9]]])