From f9b8c654cf6fb906e776463f353c04551392bd90 Mon Sep 17 00:00:00 2001 From: pengzirong Date: Fri, 1 Jul 2022 09:27:05 +0800 Subject: [PATCH] =?UTF-8?q?torch.cdist=E6=94=AF=E6=8C=81=E5=BA=A6=E4=BF=AE?= =?UTF-8?q?=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.zh.md | 6 +- ...\214\201\346\270\205\345\215\225_1.5.0.md" | 344 +++++++++--------- 2 files changed, 174 insertions(+), 176 deletions(-) diff --git a/README.zh.md b/README.zh.md index b9623fa318..79570dfb87 100644 --- a/README.zh.md +++ b/README.zh.md @@ -38,10 +38,7 @@ apt-get install -y gcc==7.3.0 g++ make build-essential libssl-dev zlib1g-dev lib 获取适配昇腾AI处理器的PyTorch源代码（即当前仓库代码），并切换到所需的分支。 ``` - git clone https://gitee.com/ascend/pytorch.git - # 当前master分支为pytorch 1.8.1版本，需要1.5.0版本请使用git checkout 命令切换到v1.5.0对应版本分支 - cd pytorch - git checkout -b v1.5.0 remotes/origin/v1.5.0 + git clone -b v1.5.0 https://gitee.com/ascend/pytorch.git ``` ## 获取原生PyTorch源代码和third_party代码 @@ -49,6 +46,7 @@ apt-get install -y gcc==7.3.0 g++ make build-essential libssl-dev zlib1g-dev lib 在当前仓库根目录pytorch/下获取原生PyTorch1.5.0的源代码。请关注PyTorch原生社区的安全板块与Issue板块是否有安全相关问题修复，并根据社区修复及时更新原生PyTorch代码。 ```sh +cd pytorch git clone -b v1.5.0 --depth=1 https://github.com/pytorch/pytorch.git ``` diff --git "a/docs/zh/PyTorch API\346\224\257\346\214\201\346\270\205\345\215\225_1.5.0.md" "b/docs/zh/PyTorch API\346\224\257\346\214\201\346\270\205\345\215\225_1.5.0.md" index 89b5c8f358..6c3f7b28db 100644 --- "a/docs/zh/PyTorch API\346\224\257\346\214\201\346\270\205\345\215\225_1.5.0.md" +++ "b/docs/zh/PyTorch API\346\224\257\346\214\201\346\270\205\345\215\225_1.5.0.md" @@ -106,178 +106,178 @@ ## Math operations -| 序号 | API名称 | 是否支持 | -| ---- | ------------------------ | ------------------------ | -| 1 | torch.abs | 是 | -| 2 | torch.acos | 是 | -| 3 | torch.add | 是 | -| 4 | torch.addcdiv | 是 | -| 5 | torch.addcmul | 是 | -| 6 | torch.angle | 否 | -| 7 | torch.asin | 是 | -| 8 | torch.atan | 是 | -| 9 | torch.atan2 | 是 | -| 10 | torch.bitwise_not | 是 | -| 11 | torch.bitwise_and | 是 | -| 12 | torch.bitwise_or | 是 | -| 13 | torch.bitwise_xor | 是 | -| 14 | torch.ceil | 是 | -| 15 | torch.clamp | 是 | -| 16 | torch.conj | 否 | -| 17 | torch.cos | 是 | -| 18 | torch.cosh | 是 | -| 19 | torch.div | 是 | -| 20 | torch.digamma | 否 | -| 21 | torch.erf | 是 | -| 22 | torch.erfc | 是 | -| 23 | torch.erfinv | 是 | -| 24 | torch.exp | 是 | -| 25 | torch.expm1 | 是 | -| 26 | torch.floor | 是 | -| 27 | torch.floor_divide | 是 | -| 28 | torch.fmod | 是 | -| 29 | torch.frac | 是 | -| 30 | torch.imag | 否 | -| 31 | torch.lerp | 是 | -| 32 | torch.lgamma | 否 | -| 33 | torch.log | 是 | -| 34 | torch.log10 | 是 | -| 35 | torch.log1p | 是 | -| 36 | torch.log2 | 是 | -| 37 | torch.logical_and | 是 | -| 38 | torch.logical_not | 是 | -| 39 | torch.logical_or | 是 | -| 40 | torch.logical_xor | 是 | -| 41 | torch.mul | 是 | -| 42 | torch.mvlgamma | 否 | -| 43 | torch.neg | 是 | -| 44 | torch.polygamma | 否 | -| 45 | torch.pow | 是 | -| 46 | torch.real | 是 | -| 47 | torch.reciprocal | 是 | -| 48 | torch.remainder | 是 | -| 49 | torch.round | 是 | -| 50 | torch.rsqrt | 是 | -| 51 | torch.sigmoid | 是 | -| 52 | torch.sign | 是 | -| 53 | torch.sin | 是 | -| 54 | torch.sinh | 是 | -| 55 | torch.sqrt | 是 | -| 56 | torch.square | 是 | -| 57 | torch.tan | 是 | -| 58 | torch.tanh | 是 | -| 59 | torch.true_divide | 是 | -| 60 | torch.trunc | 是 | -| 61 | torch.argmax | 是 | -| 62 | torch.argmin | 是 | -| 63 | torch.dist | 是 | -| 64 | torch.logsumexp | 是 | -| 65 | torch.mean | 是 | -| 66 | torch.median | 是 | -| 67 | torch.mode | 否 | -| 68 | torch.norm | 是 | -| 69 | torch.prod | 是 | -| 70 | torch.std | 是 | -| 71 | torch.std_mean | 是 | -| 72 | torch.sum | 是 | -| 73 | torch.unique | 是 | -| 74 | torch.unique_consecutive | 否 | -| 75 | torch.var | 否 | -| 76 | torch.var_mean | 否 | -| 77 | torch.allclose | 是 | -| 78 | torch.argsort | 是 | -| 79 | torch.eq | 是 | -| 80 | torch.equal | 是 | -| 81 | torch.ge | 是 | -| 82 | torch.gt | 是 | -| 83 | torch.isfinite | 是 | -| 84 | torch.isinf | 是 | -| 85 | torch.isnan | 是 | -| 86 | torch.kthvalue | 是 | -| 87 | torch.le | 是 | -| 88 | torch.lt | 是 | -| 89 | torch.max | 是 | -| 90 | torch.min | 是 | -| 91 | torch.ne | 是 | -| 92 | torch.sort | 是 | -| 93 | torch.topk | 是 | -| 94 | torch.fft | 否 | -| 95 | torch.ifft | 否 | -| 96 | torch.rfft | 否 | -| 97 | torch.irfft | 否 | -| 98 | torch.stft | 否 | -| 99 | torch.bartlett_window | 是 | -| 100 | torch.blackman_window | 是 | -| 101 | torch.hamming_window | 是 | -| 102 | torch.hann_window | 是 | -| 103 | torch.bincount | 是 | -| 104 | torch.broadcast_tensors | 是 | -| 105 | torch.cartesian_prod | 是 | -| 106 | torch.cdist | 是 | -| 107 | torch.combinations | 否 | -| 108 | torch.cross | 是 | -| 109 | torch.cummax | 是 | -| 110 | torch.cummin | 是 | -| 111 | torch.cumprod | 是 | -| 112 | torch.cumsum | 是 | -| 113 | torch.diag | 是，仅支持diagonal=0场景 | -| 114 | torch.diag_embed | 是 | -| 115 | torch.diagflat | 是 | -| 116 | torch.diagonal | 是 | -| 117 | torch.einsum | 是 | -| 118 | torch.flatten | 是 | -| 119 | torch.flip | 是 | -| 120 | torch.rot90 | 是 | -| 121 | torch.histc | 否 | -| 122 | torch.meshgrid | 是 | -| 123 | torch.renorm | 是 | -| 124 | torch.repeat_interleave | 是 | -| 125 | torch.roll | 是 | -| 126 | torch.tensordot | 是 | -| 127 | torch.trace | 否 | -| 128 | torch.tril | 是 | -| 129 | torch.tril_indices | 是 | -| 130 | torch.triu | 是 | -| 131 | torch.triu_indices | 是 | -| 132 | torch.addbmm | 是 | -| 133 | torch.addmm | 是 | -| 134 | torch.addmv | 是 | -| 135 | torch.addr | 是 | -| 136 | torch.baddbmm | 是 | -| 137 | torch.bmm | 是 | -| 138 | torch.chain_matmul | 是 | -| 139 | torch.cholesky | 否 | -| 140 | torch.cholesky_inverse | 否 | -| 141 | torch.cholesky_solve | 否 | -| 142 | torch.dot | 是 | -| 143 | torch.eig | 否 | -| 144 | torch.geqrf | 否 | -| 145 | torch.ger | 是 | -| 146 | torch.inverse | 是 | -| 147 | torch.det | 否 | -| 148 | torch.logdet | 否 | -| 149 | torch.slogdet | 是 | -| 150 | torch.lstsq | 否 | -| 151 | torch.lu | 否 | -| 152 | torch.lu_solve | 否 | -| 153 | torch.lu_unpack | 否 | -| 154 | torch.matmul | 是 | -| 155 | torch.matrix_power | 是 | -| 156 | torch.matrix_rank | 是 | -| 157 | torch.mm | 是 | -| 158 | torch.mv | 是 | -| 159 | torch.orgqr | 否 | -| 160 | torch.ormqr | 否 | -| 161 | torch.pinverse | 是 | -| 162 | torch.qr | 是 | -| 163 | torch.solve | 否 | -| 164 | torch.svd | 是 | -| 165 | torch.svd_lowrank | 是 | -| 166 | torch.pca_lowrank | 是 | -| 167 | torch.symeig | 是 | -| 168 | torch.lobpcg | 否 | -| 169 | torch.trapz | 是 | -| 170 | torch.triangular_solve | 是 | +| 序号 | API名称 | 是否支持 | +| ---- | ------------------------ | ------------------------------------------- | +| 1 | torch.abs | 是 | +| 2 | torch.acos | 是 | +| 3 | torch.add | 是 | +| 4 | torch.addcdiv | 是 | +| 5 | torch.addcmul | 是 | +| 6 | torch.angle | 否 | +| 7 | torch.asin | 是 | +| 8 | torch.atan | 是 | +| 9 | torch.atan2 | 是 | +| 10 | torch.bitwise_not | 是 | +| 11 | torch.bitwise_and | 是 | +| 12 | torch.bitwise_or | 是 | +| 13 | torch.bitwise_xor | 是 | +| 14 | torch.ceil | 是 | +| 15 | torch.clamp | 是 | +| 16 | torch.conj | 否 | +| 17 | torch.cos | 是 | +| 18 | torch.cosh | 是 | +| 19 | torch.div | 是 | +| 20 | torch.digamma | 否 | +| 21 | torch.erf | 是 | +| 22 | torch.erfc | 是 | +| 23 | torch.erfinv | 是 | +| 24 | torch.exp | 是 | +| 25 | torch.expm1 | 是 | +| 26 | torch.floor | 是 | +| 27 | torch.floor_divide | 是 | +| 28 | torch.fmod | 是 | +| 29 | torch.frac | 是 | +| 30 | torch.imag | 否 | +| 31 | torch.lerp | 是 | +| 32 | torch.lgamma | 否 | +| 33 | torch.log | 是 | +| 34 | torch.log10 | 是 | +| 35 | torch.log1p | 是 | +| 36 | torch.log2 | 是 | +| 37 | torch.logical_and | 是 | +| 38 | torch.logical_not | 是 | +| 39 | torch.logical_or | 是 | +| 40 | torch.logical_xor | 是 | +| 41 | torch.mul | 是 | +| 42 | torch.mvlgamma | 否 | +| 43 | torch.neg | 是 | +| 44 | torch.polygamma | 否 | +| 45 | torch.pow | 是 | +| 46 | torch.real | 是 | +| 47 | torch.reciprocal | 是 | +| 48 | torch.remainder | 是 | +| 49 | torch.round | 是 | +| 50 | torch.rsqrt | 是 | +| 51 | torch.sigmoid | 是 | +| 52 | torch.sign | 是 | +| 53 | torch.sin | 是 | +| 54 | torch.sinh | 是 | +| 55 | torch.sqrt | 是 | +| 56 | torch.square | 是 | +| 57 | torch.tan | 是 | +| 58 | torch.tanh | 是 | +| 59 | torch.true_divide | 是 | +| 60 | torch.trunc | 是 | +| 61 | torch.argmax | 是 | +| 62 | torch.argmin | 是 | +| 63 | torch.dist | 是 | +| 64 | torch.logsumexp | 是 | +| 65 | torch.mean | 是 | +| 66 | torch.median | 是 | +| 67 | torch.mode | 否 | +| 68 | torch.norm | 是 | +| 69 | torch.prod | 是 | +| 70 | torch.std | 是 | +| 71 | torch.std_mean | 是 | +| 72 | torch.sum | 是 | +| 73 | torch.unique | 是 | +| 74 | torch.unique_consecutive | 否 | +| 75 | torch.var | 否 | +| 76 | torch.var_mean | 否 | +| 77 | torch.allclose | 是 | +| 78 | torch.argsort | 是 | +| 79 | torch.eq | 是 | +| 80 | torch.equal | 是 | +| 81 | torch.ge | 是 | +| 82 | torch.gt | 是 | +| 83 | torch.isfinite | 是 | +| 84 | torch.isinf | 是 | +| 85 | torch.isnan | 是 | +| 86 | torch.kthvalue | 是 | +| 87 | torch.le | 是 | +| 88 | torch.lt | 是 | +| 89 | torch.max | 是 | +| 90 | torch.min | 是 | +| 91 | torch.ne | 是 | +| 92 | torch.sort | 是 | +| 93 | torch.topk | 是 | +| 94 | torch.fft | 否 | +| 95 | torch.ifft | 否 | +| 96 | torch.rfft | 否 | +| 97 | torch.irfft | 否 | +| 98 | torch.stft | 否 | +| 99 | torch.bartlett_window | 是 | +| 100 | torch.blackman_window | 是 | +| 101 | torch.hamming_window | 是 | +| 102 | torch.hann_window | 是 | +| 103 | torch.bincount | 是 | +| 104 | torch.broadcast_tensors | 是 | +| 105 | torch.cartesian_prod | 是 | +| 106 | torch.cdist | 是，仅支持mode=donot_use_mm_for_euclid_dist | +| 107 | torch.combinations | 否 | +| 108 | torch.cross | 是 | +| 109 | torch.cummax | 是 | +| 110 | torch.cummin | 是 | +| 111 | torch.cumprod | 是 | +| 112 | torch.cumsum | 是 | +| 113 | torch.diag | 是，仅支持diagonal=0场景 | +| 114 | torch.diag_embed | 是 | +| 115 | torch.diagflat | 是 | +| 116 | torch.diagonal | 是 | +| 117 | torch.einsum | 是 | +| 118 | torch.flatten | 是 | +| 119 | torch.flip | 是 | +| 120 | torch.rot90 | 是 | +| 121 | torch.histc | 否 | +| 122 | torch.meshgrid | 是 | +| 123 | torch.renorm | 是 | +| 124 | torch.repeat_interleave | 是 | +| 125 | torch.roll | 是 | +| 126 | torch.tensordot | 是 | +| 127 | torch.trace | 否 | +| 128 | torch.tril | 是 | +| 129 | torch.tril_indices | 是 | +| 130 | torch.triu | 是 | +| 131 | torch.triu_indices | 是 | +| 132 | torch.addbmm | 是 | +| 133 | torch.addmm | 是 | +| 134 | torch.addmv | 是 | +| 135 | torch.addr | 是 | +| 136 | torch.baddbmm | 是 | +| 137 | torch.bmm | 是 | +| 138 | torch.chain_matmul | 是 | +| 139 | torch.cholesky | 否 | +| 140 | torch.cholesky_inverse | 否 | +| 141 | torch.cholesky_solve | 否 | +| 142 | torch.dot | 是 | +| 143 | torch.eig | 否 | +| 144 | torch.geqrf | 否 | +| 145 | torch.ger | 是 | +| 146 | torch.inverse | 是 | +| 147 | torch.det | 否 | +| 148 | torch.logdet | 否 | +| 149 | torch.slogdet | 是 | +| 150 | torch.lstsq | 否 | +| 151 | torch.lu | 否 | +| 152 | torch.lu_solve | 否 | +| 153 | torch.lu_unpack | 否 | +| 154 | torch.matmul | 是 | +| 155 | torch.matrix_power | 是 | +| 156 | torch.matrix_rank | 是 | +| 157 | torch.mm | 是 | +| 158 | torch.mv | 是 | +| 159 | torch.orgqr | 否 | +| 160 | torch.ormqr | 否 | +| 161 | torch.pinverse | 是 | +| 162 | torch.qr | 是 | +| 163 | torch.solve | 否 | +| 164 | torch.svd | 是 | +| 165 | torch.svd_lowrank | 是 | +| 166 | torch.pca_lowrank | 是 | +| 167 | torch.symeig | 是 | +| 168 | torch.lobpcg | 否 | +| 169 | torch.trapz | 是 | +| 170 | torch.triangular_solve | 是 | ## Utilities -- Gitee