pytorch小技巧