TypeError: masked_fill() received an invalid combination of arguments - got (bool, float), but expected one of:
- (Tensor mask, Tensor value)
didn't match because some of the arguments have invalid types: (bool, float)
- (Tensor mask, Number value)
didn't match because some of the arguments have invalid types: (bool, float)