#include
#define cluster_(NAME) TH_CONCAT_4(cluster_, NAME, _, Real)
void cluster_random(THLongTensor *output, THLongTensor *row, THLongTensor *col, THLongTensor *degree) {
/* int64_t *output_data = output->storage->data + output->storageOffset; */
/* int64_t *row_data = row->storage->data + row->storageOffset; */
/* int64_t *col_data = col->storage->data + col->storageOffset; */
/* int64_t e, E = THLongTensor_nElement(row), r, c, value; */
/* for (e = 0; e < E; e++) { */
/* r = row_data[e]; c = col_data[e]; */
/* if (output_data[r] == -1 && output_data[c] == -1) { */
/* value = r < c ? r : c; */
/* output_data[r] = value; */
/* output_data[c] = value; */
/* } */
/* } */
}
#include "generic/cpu.c"
#include "THGenerateAllTypes.h"
|