pytorch tensor division/除法

版权声明：本文为博主原创文章，未经博主允许不得转载。 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)