Performs 3D average pooling on the input. Arguments: input: Shape `[batch, depth, rows, cols, channels]` tensor to pool over. ksize: 1-D tensor of length 5. The size of the window for each dimension of the input tensor. Must have `ksize[0] = ksize[4] = 1`. strides: 1-D tensor of length 5. The st
(scope *Scope, input tf.Output, ksize []int64, strides []int64, padding string, optional ...AvgPool3DAttr)
| 26643 | // |
| 26644 | // Returns The average pooled output tensor. |
| 26645 | func AvgPool3D(scope *Scope, input tf.Output, ksize []int64, strides []int64, padding string, optional ...AvgPool3DAttr) (output tf.Output) { |
| 26646 | if scope.Err() != nil { |
| 26647 | return |
| 26648 | } |
| 26649 | attrs := map[string]interface{}{"ksize": ksize, "strides": strides, "padding": padding} |
| 26650 | for _, a := range optional { |
| 26651 | a(attrs) |
| 26652 | } |
| 26653 | opspec := tf.OpSpec{ |
| 26654 | Type: "AvgPool3D", |
| 26655 | Input: []tf.Input{ |
| 26656 | input, |
| 26657 | }, |
| 26658 | Attrs: attrs, |
| 26659 | } |
| 26660 | op := scope.AddOperation(opspec) |
| 26661 | return op.Output(0) |
| 26662 | } |
| 26663 | |
| 26664 | // Gets the next output from the given iterator as an Optional variant. |
| 26665 | func IteratorGetNextAsOptional(scope *Scope, iterator tf.Output, output_types []tf.DataType, output_shapes []tf.Shape) (optional tf.Output) { |