tf.contrib.kfac.utils.extract_pointwise_conv2d_patches

tf.contrib.kfac.utils.extract_pointwise_conv2d_patches(
    inputs,
    filter_shape,
    name=None,
    data_format=None
)

Defined in tensorflow/contrib/kfac/python/ops/utils.py.

Extract patches for a 1x1 conv2d.

Args:

• inputs: 4-D Tensor of shape [batch_size, height, width, in_channels].
• filter_shape: List of 4 ints. Shape of filter to apply with conv2d()
• name: None or str. Name for Op.
• data_format: None or str. Format for data. See 'data_format' in tf.nn.conv2d() for details.

Returns:

Tensor of shape [batch_size, ..spatial_input_shape.., ..spatial_filter_shape.., in_channels]

Raises:

• ValueError: if inputs is not 4-D.
• ValueError: if filter_shape is not [1, 1, ?, ?]
• ValueError: if data_format is not channels-last.