pytorch 与numpy 部分操作的对应关系