Open kachayev opened 2 months ago
Right now we assume that features (e.g. features_s, features_t) are 2-dimentional tensors. This might not always be the case. Should be either stated in the documentation clearly, or we can make it work for ndim > 2.
features_s
features_t
Right now we assume that features (e.g.
features_s
,features_t
) are 2-dimentional tensors. This might not always be the case. Should be either stated in the documentation clearly, or we can make it work for ndim > 2.