0
点赞
收藏
分享

微信扫一扫

关于springboot创建kafkaTopic

_刘彦辉 2024-06-25 阅读 29

如果你有两个形状为 (3, 4) 的tensor A 和 B,并且你想要在第一个维度上连接它们,你可以这样做:

import torch
 
A = torch.randn(3, 4)
B = torch.randn(3, 4)
C = torch.cat([A, B], dim=0)  # dim=0 表示在第一个维度上连接

此时 C 的形状将会是 (6, 4),因为你在第一个维度上加入了 A 和 B 的元素数量。

C = torch.cat([A, B], dim=1)  # dim=1 表示在第二个维度上连接

此时 C 的形状将会是 (3, 8),因为在第二个维度上加入了 A 和 B 的元素数量。

举报

相关推荐

0 条评论