QuadrupleCartesianProd

class pinnx.problem.QuadrupleCartesianProd(X_train, y_train, X_test, y_test)

Cartesian Product input data format for MIONet architecture.

This dataset can be used with the network MIONetCartesianProd for operator learning.

Parameters:

• X_train – A tuple of three NumPy arrays. The first element has the shape (N1, dim1), the second element has the shape (N1, dim2), and the third element has the shape (N2, dim3).

• y_train – A NumPy array of shape (N1, N2).

losses(inputs, outputs, targets, **kwargs)

Return a list of losses, i.e., constraints.

test()

Return a test dataset.

train_next_batch(batch_size=None)

Return a training dataset of the size batch_size.