I want to know a scalar is equal to one or not. How to write such code ?
scalar = torch.tensor(1)
assert scalar==1
# or
if scalar==1:
print('yes')
It will report an error
—Original—
I want to know a scalar is equal to one or not. How to write such code ?
scalar = torch.tensor(1)
assert scalar==1
# or
if scalar==1:
print('yes')
It will report an error
—Original—