torch – 编程知识网

首页 › 文章标签 "torch"

torch基础学习

1.PyTorch 中，nn 与 nn.functional 有什么区别？这两个是差不多的，不过一个包装好的类，一个是可以直接调用的函数。 ...

风君子
软件
2024-03-31
13

torch入门

张量的创建 import torch import numpy as np#创建一个张量 x=torch.randn((5,3),dtype=torch.float16) #张量的 ...

风君子
软件
2023-11-18
35

diou ciou torch

ciou diou torch

diou ciou torch

diou比ciou要小，比iou的值也小，训练时候用比较好。 ...

风君子
软件
2023-11-10
23

torch中的mul(、matmul(和mm(

torch中的mul(、matmul(和mm(

1.mul() 源码中是这样的。可以看到主要有两个参数是必要的，分别是两个tensor向量。 ...

风君子
后端
2023-05-14
28