pytorch 数据操作技巧

1. a

import torch
align = torch.FloatTensor([3,4,8]).to(torch.long)
torch.repeat_interleave(torch.eye(3),align,dim=1)

tensor([[1., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],
[0., 0., 0., 1., 1., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0.],
[0., 0., 0., 0., 0., 0., 0., 1., 1., 1., 1., 1., 1., 1., 1.]])

标签：dim,repeat,eye,技巧,align,torch,long,pytorch,操作
来源： https://blog.csdn.net/zz_hh_uu_/article/details/120683804