U = U_w + U_j # b*c*es*1
print("concat:", U.shape)
p = torch.mean(U, dim=-1, out=None) # b*c*es
print(Ugp.shape)

U = U_w + U_j # b*c*es*1
print("concat:", U.shape)
p = torch.mean(U, dim=-1, out=None) # b*c*es
print(Ugp.shape)

您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
2万+
1899

被折叠的 条评论
为什么被折叠?