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 |