我有以下张量:
vector = torch.tensor([[1,5,3], [2,3,4]])
Run Code Online (Sandbox Code Playgroud)
如何交换第二个轴中的值?例如
tensor([[1, 5, 3], [2, 3, 4]])
变成:
tensor([[1, 3, 3], [2, 5, 4]])
我有两个数组(Int32)列,有什么方法可以比较它们并获取这两个数组的交集的列?
我正在考虑 arrayMap 函数,但这似乎是无效的方法。
示例数据:
id col1 col2 col3
1 [1, 2, 3] [2, 3, 5] [2, 3]
...........
Run Code Online (Sandbox Code Playgroud)