已合并
MatrixSetDiag修改proto描述 #1514
onanfield创建于 3月5日
MatrixSetDiag修改proto描述 #1514
已合并
共 1 个文件变更+6-4
| @@ -20,13 +20,15 @@ namespace ge { | |||
| 20 | 20 | ||
| 21 | * @par Inputs: | 21 | * @par Inputs: |
| 22 | * Two inputs, including: | 22 | * Two inputs, including: |
| 23 | -* @li x: A Tensor. Must be one of the following types: | 23 | +* @li x: A Tensor. 2-8 dimensions. Must be one of the following types: |
| 24 | * float16, float32, double, int32, uint8, int16, int8, complex64, int64, | 24 | * float16, float32, double, int32, uint8, int16, int8, complex64, int64, |
| 25 | -* qint8, quint8, qint32, uint16, complex128, uint32, uint64. | 25 | +* qint8, quint8, qint32, uint16, complex128, uint32, uint64, bfloat16, bool. |
| 26 | -* @li diagonal: A Tensor of the same type as "x" . \n | 26 | +* @li diagonal: A Tensor of the same type as "x". 1-7 dimensions (must be one less than x). |
| 27 | + The last dimension is the smaller of the last two dimensions of x, | ||
| 28 | + while the other dimensions are same as those in x. \n | ||
| 27 | 29 | ||
| 28 | * @par Outputs: | 30 | * @par Outputs: |
| 29 | -* y: A Tensor. Has the same type as "x" . \n | 31 | +* y: A Tensor. Has the same type and shape as "x" . \n |
| 30 | 32 | ||
| 31 | * @par Third-party framework compatibility | 33 | * @par Third-party framework compatibility |
| 32 | * Compatible with the TensorFlow operator MatrixSetDiag. | 34 | * Compatible with the TensorFlow operator MatrixSetDiag. |