msnh2012/Msnhnet

pytorch的reshape 不是torch.Tensor下的op

lsydd opened this issue · 1 comments

lsydd commented

应该直接用

raw_reshape = torch.reshape
torch.reshape = _reshape

进行hook.

两种都可以, 多谢issue