张量的创建 import torch import numpy as np#创建一个张量 x=torch.randn((5,3),dtype=torch.float16) #张量的 … Continue reading torch入门
标签: torch
diou ciou torch
diou比ciou要小,比iou的值也小,训练时候用比较好。 测试时还是iou比较好,训练时giou比较好 import math import torchdef bbox_iou(box1, box … Continue reading diou ciou torch
Torch 的安装与基本用法
本文安装仅限 ubuntu 系统。官方文档见:Getting started with Torch。 0. 简介 Torch 使用轻量级脚本语言 Lua 及其 C/CUDA 扩展模块实现 … Continue reading Torch 的安装与基本用法
torch五项(pytorch handbook
如果您对机器学习感兴趣,或者正在阅读这篇文章,您很可能以前听过PyTorch。 在机器学习开发者中,这是一个非常有名的框架,为了好的理由,本文进行讨论。 另外,如果只是从神经网络开始掌握基础知识,这是 … Continue reading torch五项(pytorch handbook
torch中的mul(、matmul(和mm(
1.mul() 源码中是这样的。可以看到主要有两个参数是必要的,分别是两个tensor向量。 def mul(input: Union[Tensor, Number], other: U … Continue reading torch中的mul(、matmul(和mm(