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)
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
