output = torch.matmul(emb_item[data.batch == i], x[i,:]). outputs.append(output). x = torch.cat(outputs, dim=0). x = torch.sigmoid(x).
確定! 回上一頁