Inputs of the SetTransformer

[SRL94]

Hi,

Could you please explain the meanings of the inputs of SetTransformer:

dim_input, num_outputs, dim_output, num_inds=32, dim_hidden=128, num_heads=4, ln=False

Thanks.

[juho-lee]

I think the names of the variables pretty much explain everything. Dimension of the inputs, number of outputs (number of seeds for output PMA), dimension of the outputs, number of inducing points, and so on.