pymiediff.Particle.set_device

Particle.set_device(device)

Move all stored tensors/materials to a new torch device.

Parameters:

device (str or torch.device) – Target device.