torch.gather(input, dim, index) -> Tensor
input和index必须维度数目相同,除维度dim外其他维度d上,index的size小于等于input的 out和indexshape相同,
torch.arange(start, end, step)
不含end 返回1d tensor
tensor.repeat(a,b,c,,,,)
参数个数和tensor维度数目相同,相应维度上重复相应次
torch.nn.utils.rnn.pad_sequence(sequences, batch_first=False, padding_value=0.0)
sequences是list,元素是tensor,batch_first只与输出有关 按tensor最长pad, 长度是tensor的第0维