def updateOutput(self, input):
input, mask = input
torch.masked_select(input, mask, out=self.output)
return self.output
微信
公众号
扫码关注公众号