我有一个Tensor：
tensor([[ 9, -1, -1], [ 7, -1, -1], [ 6, 4, -1]])
什么是最有效的方法来删除填充，并得到这样的东西：
[[9], [7], [ 6, 4]])
提前感谢您所能提供的任何帮助。
hkmswyz61#
请考虑在Pytorch中使用PackedSequence。它适用于具有可变长度序列的批处理，并受Pytorch RNN单元格支持。您可以使用torch.nn.utils.rnn.pack_padded_sequence（）方法创建PackedSequence。请将padding_value指定为-1，因为您正在使用-1作为填充。
padding_value
1条答案
按热度按时间hkmswyz61#
请考虑在Pytorch中使用PackedSequence。它适用于具有可变长度序列的批处理，并受Pytorch RNN单元格支持。您可以使用torch.nn.utils.rnn.pack_padded_sequence（）方法创建PackedSequence。请将
padding_value指定为-1，因为您正在使用-1作为填充。