torch用法记录(未完结)

161 阅读1分钟

torch.gather(input, dim, index) -> Tensor

image.png

input和index必须维度数目相同,除维度dim外其他维度d上,index的size小于等于input的 out和indexshape相同,

image.png

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维