* change np.transpose to torch.permute * add comments
Attach a file by drag & drop or click to upload
