I have some issue understanding what is inp and targ in the metrics
also in the documentation is mentioned “out” shouldn’t be inp?
#export torch_core
def flatten_check(inp, targ):
"Check that `out` and `targ` have the same number of elements and flatten them."
inp,targ = inp.contiguous().view(-1),targ.contiguous().view(-1)
test_eq(len(inp), len(targ))
return inp,targ