flatten_check {fastai}

Flatten check

Description

Check that 'out' and 'targ' have the same number of elements and flatten them.

Usage

flatten_check(inp, targ)

Arguments

inp

predictions

targ

targets

Value

tensor