版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_29007291/article/details/84529506
除法的时候要注意数据类型的问题
a = torch.div(torch.tensor([160, 110]), 0.137)
jupyter notebook就会奔溃, 因为torch.tensor([160, 110])
是torch.int64
数据类型, 而0.137(或者说torch.tensor(0.137))是torch.float32类型.
如果改成下面这样, 就会顺利运行
a = torch.div(torch.tensor([160, 110]).float(), 0.137)