compute_conf_mat#

compute_conf_mat(out: Tensor, y: Tensor, num_labels: int) → Tensor[source]#

Parameters

out (Tensor) –
y (Tensor) –
num_labels (int) –

Return type