emit_squeeze_node Subroutine

public subroutine emit_squeeze_node(name, input, axes_input, output, nodes, num_nodes)

Emit a Squeeze node (ONNX opset 13+: axes as input)

Arguments

Type IntentOptional Attributes Name
character(len=*), intent(in) :: name

Node name, data input, axes input, and output tensor name
character(len=*), intent(in) :: input

Node name, data input, axes input, and output tensor name
character(len=*), intent(in) :: axes_input

Node name, data input, axes input, and output tensor name
character(len=*), intent(in) :: output

Node name, data input, axes input, and output tensor name
type(onnx_node_type), intent(inout), dimension(:) :: nodes

Node accumulator array
integer, intent(inout) :: num_nodes

Current number of populated nodes

Source Code

  subroutine emit_squeeze_node(name, input, axes_input, output, &
       nodes, num_nodes)
    !! Emit a Squeeze node (ONNX opset 13+: axes as input)
    implicit none

    ! Arguments
    character(*), intent(in) :: name, input, axes_input, output
    !! Node name, data input, axes input, and output tensor name
    type(onnx_node_type), intent(inout), dimension(:) :: nodes
    !! Node accumulator array
    integer, intent(inout) :: num_nodes
    !! Current number of populated nodes

    call emit_node('Squeeze', name, output, '', nodes, num_nodes, &
         in1=input, in2=axes_input)

  end subroutine emit_squeeze_node