Electro Magnetic fields;electromagnetic devices;deep learning

For coil dataset，the shape of one data is [160,160,5]

For transformer dataset，the shape of one data is [400,400,5]

For IPM motor dataset，the shape of one data is [180,180,5]

The first five layers consist of input device property information, while the sixth layer represents the FEA magnetic field distribution results.

When using, please note to split the dataset into two parts, with x being transformed as follows:

X = data[:, :, :5]
X = np.transpose(X, (2, 0, 1))
X = torch.from_numpy(X)

, and y being transformed as follows: