Comment Comment by master_yoda_1 I would better use pytorch api than this bogus code. https://pytorch.org/docs/stable/generated/torch.nn.Transform... parent master_yoda_1 3 years ago
