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 |