From 0a1449a983e6e88e62e175d0f34564414d26c4dd Mon Sep 17 00:00:00 2001 From: Andrew Selle Date: Tue, 26 May 2020 20:51:52 -0700 Subject: [PATCH] Fix bugs in docstring of conv3d_transpose. Seems like depth was missing and NCHW and NHWC were mentioned instead of NCDHW and NDHWC. PiperOrigin-RevId: 313320657 Change-Id: Id599e6e7b91a18f193b952153c50c27839c5693f --- tensorflow/python/ops/nn_ops.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tensorflow/python/ops/nn_ops.py b/tensorflow/python/ops/nn_ops.py index 24ee94fac48..b7dd1d20aae 100644 --- a/tensorflow/python/ops/nn_ops.py +++ b/tensorflow/python/ops/nn_ops.py @@ -3061,12 +3061,12 @@ def conv3d_transpose_v2(input, # pylint: disable=redefined-builtin rather than an actual deconvolution. Args: - input: A 5-D `Tensor` of type `float` and shape `[batch, height, width, - in_channels]` for `NHWC` data format or `[batch, in_channels, height, - width]` for `NCHW` data format. - filters: A 5-D `Tensor` with the same type as `value` and shape `[height, - width, output_channels, in_channels]`. `filter`'s `in_channels` dimension - must match that of `value`. + input: A 5-D `Tensor` of type `float` and shape `[batch, depth, height, + width, in_channels]` for `NDHWC` data format or `[batch, in_channels, + depth, height, width]` for `NCDHW` data format. + filters: A 5-D `Tensor` with the same type as `value` and shape `[depth, + height, width, output_channels, in_channels]`. `filter`'s `in_channels` + dimension must match that of `value`. output_shape: A 1-D `Tensor` representing the output shape of the deconvolution op. strides: An int or list of `ints` that has length `1`, `3` or `5`. The