【易踩坑】torch.split函数

发布网友发布时间：2023-07-08 18:51

共1个回答

热心网友时间：2023-10-24 06:45

torch.split(tensor, split_size, dim=)

tensor是要切割的张量，dim表示在哪个维度上面进行切割

a = torch.LongTensor([[1,2,3,4],[2,3,4,5]])

b = torch.cat(torch.split(a, 4, dim=1), dim=0)

print(b)

输出：tensor([[1, 2, 3, 4],

[2, 3, 4, 5]])

声明：本网页内容为用户发布，旨在传播知识，不代表本网认同其观点，若有侵权等问题请及时与本网联系，我们将在第一时间删除处理。E-MAIL:11247931@qq.com