Overloaded functions

The following pytorch functions can be called with pytorch_ndonnx.Tensor objects.

• torch.add
• torch.argmax
• torch.argmin
• torch.cat
• torch.chunk
• torch.flatten
• torch.nn.functional.adaptive_avg_pool2d
• torch.nn.functional.avg_pool2d
• torch.nn.functional.batch_norm
• torch.nn.functional.conv2d
• torch.nn.functional.conv_transpose2d
• torch.nn.functional.dropout
• torch.nn.functional.dropout2d
• torch.nn.functional.embedding
• torch.nn.functional.gelu
• torch.nn.functional.hardtanh
• torch.nn.functional.layer_norm
• torch.nn.functional.leaky_relu
• torch.nn.functional.linear
• torch.nn.functional.log_softmax
• torch.nn.functional.max_pool2d
• torch.nn.functional.max_pool2d_with_indices (undocumented upstream)
• torch.nn.functional.pixel_shuffle
• torch.nn.functional.prelu
• torch.nn.functional.relu
• torch.nn.functional.scaled_dot_product_attention
• torch.nn.functional.sigmoid
• torch.nn.functional.softmax
• torch.nn.functional.tanh
• torch.sqrt
• torch.sub
• torch.transpose