pytorch 笛卡尔积

最新推荐文章于 2023-08-09 14:58:58 发布

Oysterman

最新推荐文章于 2023-08-09 14:58:58 发布

阅读量657

点赞数

文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/zhziqiang/article/details/130399214

版权

文章介绍了笛卡尔积的概念，并通过Python的itertools库和PyTorch张量的cartesian_prod函数展示了如何计算数组和张量的笛卡尔积。示例代码详细演示了这两个方法的使用，包括对矩阵进行笛卡尔积操作后的结果展示和解释。

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

什么是笛卡尔积？

待补充

数组的笛卡尔积

>>> a = [1, 2, 3]
>>> b = [4, 5]
>>> list(itertools.product(a, b))
[(1, 4), (1, 5), (2, 4), (2, 5), (3, 4), (3, 5)]
>>> tensor_a = torch.tensor(a)
>>> tensor_b = torch.tensor(b)
>>> torch.cartesian_prod(tensor_a, tensor_b)
tensor([[1, 4],
        [1, 5],
        [2, 4],
        [2, 5],
        [3, 4],
        [3, 5]])