CustomNorm.to_device

CustomNorm.to_device(device)

Return a version of the same weighting, but with any internal arrays stored on a different device.