Output Variable

op_output_variable(shape, dtype, dynamic_axes, needs_gradient = TRUE,
  name = "")

Arguments

shape

- list of ints representing tensor shape

dtype

- data type to be used ("float32", "float64", or "auto")

name

(str) the name of the Function instance in the network