教程中的 Torchtext BucketIterator 包装器产生 SyntaxError

Question

我正在关注并在 Torchtext 上实现来自 this short tutorial 的代码，鉴于 Torchtext 的文档不佳，这一点令人惊讶地清晰。

创建迭代器（批处理生成器）后，他建议创建一个包装器以生成更多可重用的代码。 （参见教程中的第 5 步）。

该代码包含一个令人惊讶的长而奇怪的行，我不明白它并引发了 SyntaxError: invalid syntax。有人知道发生了什么吗？

（有问题的行是开头的行：if self.y_vars is <g [...]）

class BatchWrapper:
  def __init__(self, dl, x_var, y_vars):
        self.dl, self.x_var, self.y_vars = dl, x_var, y_vars # we pass in the list of attributes for x <g class="gr_ gr_3178 gr-alert gr_spell gr_inline_cards gr_disable_anim_appear ContextualSpelling ins-del" id="3178" data-gr-id="3178">and y</g>

  def __iter__(self):
        for batch in self.dl:
              x = getattr(batch, self.x_var) # we assume only one input in this wrapper

              if self.y_vars is <g class="gr_ gr_3177 gr-alert gr_gramm gr_inline_cards gr_disable_anim_appear Grammar replaceWithoutSep" id="3177" data-gr-id="3177">not</g> None: # we will concatenate y into a single tensor
                    y = torch.cat([getattr(batch, feat).unsqueeze(1) for feat in self.y_vars], dim=1).float()
              else:
                    y = torch.zeros((1))

              yield (x, y)

  def __len__(self):
        return len(self.dl)

Answer 1

是的，我想作者有一些错字。 我认为正确的代码是这样的：

if self.y_vars is not None:
    y = torch.cat([getattr(batch, feat).unsqueeze(1) for feat in self.y_vars], dim=1).float()
else:
    y = torch.zeros((1))

您也可以在第 3 行的评论中看到这个错字（在博文中的代码中）。