def pool_max(tensor, dim):
return torch.max(tensor, dim)[0]
微信
公众号
扫码关注公众号
微信
公众号