nvidia.dali.experimental.dynamic.shapes

nvidia.dali.experimental.dynamic.shapes(input, /, *, batch_size=None, device=None, dtype=None, type=None)

Warning

This operator is now deprecated.

Use nvidia.dali.pipeline.DataNode.shape() instead.

Returns the shapes of tensors in the input batch.

This operator allows sequence inputs and supports volumetric data.

Supported backends

• ‘cpu’

• ‘gpu’

Parameters:

input (Tensor/Batch) – Input to the operator.

Keyword Arguments:

• dtype (nvidia.dali.types.DALIDataType, optional, default = DALIDataType.INT64) – Data type to which the sizes are converted.

• type (nvidia.dali.types.DALIDataType) –

  Warning

  The argument type is a deprecated alias for dtype. Use dtype instead.