def get_dense_matrix(self, data: torch.tensor, c: torch.tensor): batch_size = c[-1, -1] + 1 data = spconv.SparseConvTensor(data.unsqueeze(1), c[:, self.permute_tensor], self.spatial_size, batch_size) data = data.dense() data = data.detach().cpu().numpy() return data
def get_dense_matrix(self, data: torch.tensor, c: torch.tensor, to_numpy=True): batch_size = c[-1, -1] + 1 if data.dim() == 1: data = spconv.SparseConvTensor(data.unsqueeze(1), c[:, self.permute_tensor], self.spatial_size, batch_size) else: data = spconv.SparseConvTensor(data, c[:, self.permute_tensor], self.spatial_size, batch_size) data = data.dense() if to_numpy: data = data.detach().cpu().numpy() return data