start from importing some stuff import torch import torch.nn as nn import ... += [output] outputs = torch.stack(outputs, 1).squeeze(2) return outputs.
確定! 回上一頁