torch里面的函数

最新推荐文章于 2024-08-05 14:35:27 发布

喵喵喵喵诺

最新推荐文章于 2024-08-05 14:35:27 发布

阅读量976

点赞数

分类专栏： pytorch python

本文链接：https://blog.csdn.net/qq_42987423/article/details/109456152

版权

这篇博客介绍了PyTorch中的一些关键函数，包括torch.scatter的用法，如何在reduce为mean时进行计算，以及torch.addmm在矩阵乘法中的应用。还探讨了torch.where实现的条件选择，数组切片的实现方式，以及卷积操作和数据约束。此外，提到了torch.clamp用于限制数据范围，并通过错误示例解释了参数传递的注意事项。文章最后讨论了张量的切分和拼接操作。

摘要由CSDN通过智能技术生成

import torch
from torch_scatter import scatter
src = torch.range(1,24).view(2,6,2)
index = torch.tensor([0, 1, 0, 1, 2, 1])
# Broadcasting in the first and last dim.
out = scatter(src, index, dim=1, reduce="sum")
print(src)
print(out)

Pytorch学习（二十六）---- torch.scatter的使用

tensor([[[ 1.,  2.],
         [ 3.,  4.],
         [ 5.,  6.],
         [ 7.,  8.],
         [ 9., 10.],
         [11., 12.]],

        [[13., 14.],
         [15., 16.],
         [17., 18.],
         [19.