onnx中的常见操作

具体见: onnx Operator Schemas

取一个tensor的shape

torch.tensor([[2, 3, 4], [3,4,5]])

# shape操作输出 (2, 3)

一般用来取shape的某一个维度，例如

torch.tensor([2, 256, 768]).gather(index=0)
# 输出2

torch.tensor([2, 256, 768]).gather(index=1)
# 输出256

扩充维度

将list of tensor合并成一个tensor

将一个tensor变换形状，例如

torch.tensor([1, 2, 3, 4]).reshape(2, 2)
"""
输出:
[[1, 2],
 [3, 4]]
"""

矩阵相乘

对应位置加、减、乘、除法。