cb287ae46f67324fbd4eb0a6a919673344ac1017,torchtext/data/field.py,NestedField,numericalize,#NestedField#Any#Any#Any#,589

Before Change

            numericalized_ex = self.nesting_field.numericalize(
                arr, device=device, train=train)
            numericalized.append(numericalized_ex)
        return torch.stack(numericalized)


class LabelField(Field):
    A Label field.

After Change

            numericalized_ex = self.nesting_field.numericalize(
                arr, device=device, train=train)
            numericalized.append(numericalized_ex)
        padded_batch = torch.stack(numericalized)

        self.nesting_field.include_lengths = True
        if self.include_lengths:
            sentence_lengths = torch.LongTensor(sentence_lengths)
            word_lengths = torch.LongTensor(word_lengths)
            if device == -1:
                return (padded_batch, sentence_lengths, word_lengths)
            else:
                return (padded_batch,
                        sentence_lengths.cuda(device), word_lengths.cuda(device))
        return padded_batch

Instances