如题:直接使用list函数。
例如:
noisy_code = linear_code - torch.normal(mean=0, std=args.noise_dev, size=list(linear_code.size()))
如题:直接使用list函数。
例如:
noisy_code = linear_code - torch.normal(mean=0, std=args.noise_dev, size=list(linear_code.size()))