- 博客(12)
- 收藏
- 关注
原创 MinkowskiEngine.cat和MinkowskiEngine.MinkowskiOps.cat
MinkowskiEngine.cat和MinkowskiEngine.MinkowskiOps.cat都是MinkowskiEngine库中的函数,用于级联稀疏张量。它们的作用是将多个稀疏张量连接起来,形成一个新的稀疏张量。具体来说,MinkowskiEngine.cat是一个通用的级联函数,可以接受任意数量的稀疏张量作为输入,并返回一个新的稀疏张量。而MinkowskiEngine.MinkowskiOps.cat则是一个特定于Minkowski引擎的级联函数,它只能接受在Minkowski引擎中
2023-08-04 15:07:29
317
1
原创 NCCL变量设置
这些是环境变量的设置命令,用于配置 NVIDIA Collective Communications Library (NCCL)的运行参数。NCCL_DEBUG=INFO 含义。
2023-08-02 09:58:06
4450
1
原创 torch.int8和torch.uint8
torch.int8和torch.uint8都是PyTorch中的数据类型,它们的主要差别在于。
2023-07-26 19:53:25
1158
1
原创 torch.sparse_coo_tensor
size参数表示输出张量的形状,即输出张量的行数和列数。例如,如果输入矩阵是一个4x5的矩阵,但是只有第2行和第4行、第3列和第5列上的元素是非零的,那么输出张量的形状就是(2, 2)。同时,由于只存储了非零元素的位置和值,因此可以节省大量的内存空间。torch.sparse_coo_tensor是PyTorch中的一个稀疏张量类,用于表示稀疏矩阵。indices参数表示非零元素在原矩阵中的位置,即哪些位置是非零的。它的形状为(N, 2),其中。values参数表示非零元素的值,即。它的形状为(N,)。
2023-07-26 11:22:17
1087
1
原创 torchvision.transforms.functional.hflip
该函数接受一个张量作为输入,表示要进行翻转的图像。如果输入是一个 PIL 图像,则需要先将其转换为张量。函数返回一个新的张量,其中包含水平翻转后的图像。torchvision.transforms.functional.hflip 是 PyTorch 中的一个函数,用于将输入的图像水平翻转。
2023-06-20 15:39:24
738
原创 torchvision.transforms.functional.resized_crop
torchvision.transforms.functional.resized_crop 是 PyTorch 中的一个函数,它用于从输入图像中裁剪指定大小的区域。该函数返回一个4D张量,表示裁剪后的图像,形状为 (C, size[0], size[1])。
2023-06-20 15:38:47
732
原创 torch.as_tensor()
在PyTorch中,张量是一种多维数组,可以用于表示各种类型的数据,例如数字、向量、矩阵等。torch.as_tensor()函数可以将Python对象转换为张量,并返回一个新的张量对象。其中,参数object是要转换为张量的Python对象,可以是列表、元组、NumPy数组、Pandas DataFrame或Series等。torch.as_tensor()是PyTorch中的一个函数,用于将Python对象转换为张量(Tensor)。参数dtype和device分别指定了新张量的数据类型和设备。
2023-06-20 14:35:56
1137
原创 mmdetection中的load_from和resume_from
是两个不同的函数,分别用于加载完整的模型文件和继续加载上一次未完成的训练。它们的联系在于都是用来加载模型文件的函数,但在具体的应用场景中需要根据需求选择合适的函数。都是mmdetection库中的函数,用于加载模型文件。函数来加载二进制文件和配置文件,然后将它们合并成一个完整的模型对象。函数来加载上一次保存的模型文件,并使用它来继续训练或推理。
2023-06-03 16:57:05
3036
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人