Skip to content

Commit 26fa421

Browse files
committed
add class comment
1 parent 75d68f5 commit 26fa421

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tf2onnx/onnx_opset/tensor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1823,6 +1823,7 @@ def version_11(cls, ctx, node, **kwargs):
18231823
class MatrixDiagPartV2V3:
18241824
@classmethod
18251825
def version_11(cls, ctx, node, **kwargs):
1826+
# assemble MatrixDiagPart V2&V3 by looping k diagonals with proper pads
18261827
input_tensor = node.input[0]
18271828
k = ctx.make_node('Cast', [node.input[1]], attr={'to': TensorProto.INT64}).output[0]
18281829
padding = node.input[2]

0 commit comments

Comments
 (0)