当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python tf.RaggedTensor.bounding_shape用法及代码示例


用法

bounding_shape(
    axis=None, name=None, out_type=None
)

参数

  • axis 一个整数标量或向量，指示要为其返回边界框的轴。如果未指定，则返回完整的边界框。
  • name 返回张量的名称前缀(可选)。
  • out_type dtype 用于返回的张量。默认为 self.row_splits.dtype 。

返回

  • 一个整数 Tensor ( dtype=self.row_splits.dtype )。如果未指定 axis，则 output 是带有 output.shape=[self.shape.ndims] 的向量。如果axis 是标量，则output 是标量。如果 axis 是向量，则 output 是向量，其中 output[i] 是维度 axis[i] 的边界大小。

返回此 RaggedTensor 的紧密边界框形状。

例子：

rt = tf.ragged.constant([[1, 2, 3, 4], [5], [], [6, 7, 8, 9], [10]])
rt.bounding_shape().numpy()
array([5, 4])

相关用法


注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.RaggedTensor.bounding_shape。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。