Use torch.amax in ReduceMax forward to support multiplie dim#571
Open
ice-tong wants to merge 1 commit into
Open
Use torch.amax in ReduceMax forward to support multiplie dim#571ice-tong wants to merge 1 commit into
ice-tong wants to merge 1 commit into