From 848ec6e5c66e49033eb75c1779608b910a646196 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=99=B3=E5=BB=BA=E6=88=90=EF=BC=88Chien-Cheng=20=28Jeff?=
 =?UTF-8?q?=29=20Chen=EF=BC=89?=
 <43311603+jeffeuxMartin@users.noreply.github.com>
Date: Thu, 17 Mar 2022 09:12:11 +0800
Subject: [PATCH] Fix dimension bug of feature extraction

The original code took slices on the wrong dimension.
---
 textless/data/cpc_feature_reader.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/textless/data/cpc_feature_reader.py b/textless/data/cpc_feature_reader.py
index d19abea..e39483b 100644
--- a/textless/data/cpc_feature_reader.py
+++ b/textless/data/cpc_feature_reader.py
@@ -47,7 +47,8 @@ def get_features(self, x: torch.Tensor) -> torch.Tensor:
             start += self.max_chunk
 
         if start < size:
-            x_chunk = x[:, -self.max_chunk :]
+            x_chunk = x[..., -self.max_chunk :]  # dimension wrong, \
+                                                 #     compare with above
             feat_chunk = self.model.extract_features(
                 source=x_chunk,
                 get_encoded=self.use_encoder_layer,