Pytorch中常见函数1
torch.unfold(函数:unfold(), view(), reshape(), permute(), transpose(), flatten(), cat(), chunk(), split(), stack(), take(), tile(), unbind(), squeeze(), unsqueeze(), where(), full(), cumprod(), gather()2.view()view()将tensor重塑为想要的shape。其操作是将张量展平成一维之后,再排列成
原创
2022-06-24 17:01:24 ·
570 阅读 ·
1 评论