several channels in BiMaps layers

[ygerf]

Hi,
I may be using it wrong but I have not found the way to have several channels in the BiMap layer.
In the original paper, it was designed to do so.
Would that be possible to have this option ?
Cheers (and thanks for this awesome library)

[bruAristimunha]

In this case, it would be better to use the BiMap with the depthwise option for the number of channels.

A good example of its use is in tensorcsp, where we define a bimap for each frequency component, which is separated into a dimension.

spd_learn/spd_learn/models/tensorcsp.py

Lines 102 to 111 in a9e23c3

	if layer_num > 1:
	for i in range(layer_num - 1):
	dim_in, dim_out = self.dims[2 * i], self.dims[2 * i + 1]
	layers.append(
	BiMap(
	depthwise=self.depthwise,
	in_features=dim_in,
	out_features=dim_out,
	parametrized=False,
	)