转稀疏矩阵直接求softmax
temp = self.gen_mlp(self.dropout(ff)).reshape(-1)
z_matrix = torch.sparse.FloatTensor(v_indices, temp, (num_node, num_node))
pi = torch.sparse.softmax(z_matrix, dim=1)
转稀疏矩阵直接求softmax
temp = self.gen_mlp(self.dropout(ff)).reshape(-1)
z_matrix = torch.sparse.FloatTensor(v_indices, temp, (num_node, num_node))
pi = torch.sparse.softmax(z_matrix, dim=1)