Pytorch求张量的倒数

最新推荐文章于 2023-01-20 22:56:03 发布

飞机火车巴雷特

最新推荐文章于 2023-01-20 22:56:03 发布

阅读量3k

点赞数 3

分类专栏： Python相关文章标签： pytorch python

本文链接：https://blog.csdn.net/qq_36158230/article/details/121154497

版权

Python相关专栏收录该内容

61 篇文章 1 订阅

订阅专栏

代码

import torch

# 定义两个张量
a = torch.tensor([[1.0, 2.0, 4.0],
                  [8.0, 6.0, 4.0],
                  [6.0, 8.0, 9.0],
                  [1.0, 2.0, 8.0]])
b = torch.tensor([0.8, 1.0, 0.6, 0.4])

# 求张量的倒数
r_a = 1.0 / a
r_b = 1.0 / b

# 打印
print(a)
print(b)
print(r_a)
print(r_b)

输出

tensor([[1., 2., 4.],
[8., 6., 4.],
[6., 8., 9.],
[1., 2., 8.]])
tensor([0.8000, 1.0000, 0.6000, 0.4000])
tensor([[1.0000, 0.5000, 0.2500],
[0.1250, 0.1667, 0.2500],
[0.1667, 0.1250, 0.1111],
[1.0000, 0.5000, 0.1250]])
tensor([1.2500, 1.0000, 1.6667, 2.5000])

注意：求倒数之前，要确保没有张量中有值为0.0的元素，否则会报错。