OutputListSize returns the size of the list of Outputs that is produced by a named output of op. An Operation has multiple named outputs, each of which produces either a single tensor or a list of tensors. This method returns the size of the list of tensors for a specific output of the operation, i
(output string)
| 59 | // the list of tensors for a specific output of the operation, identified |
| 60 | // by its name. |
| 61 | func (op *Operation) OutputListSize(output string) (int, error) { |
| 62 | cname := C.CString(output) |
| 63 | defer C.free(unsafe.Pointer(cname)) |
| 64 | status := newStatus() |
| 65 | n := C.TF_OperationOutputListLength(op.c, cname, status.c) |
| 66 | return int(n), status.Err() |
| 67 | } |
| 68 | |
| 69 | // Output returns the i-th output of op. |
| 70 | func (op *Operation) Output(i int) Output { |