Method shape

dali/python/nvidia/dali/data_node.py:317–331

Returns the run-time shapes of this DataNode as a new DataNode Parameters ---------- arg_dtype : DALIDataType, optional If specified, the shape will be converted to this data type; defaults to INT64. device : str, optional The device ("cpu" or

(self, *, dtype=None, device="cpu")

            return nvidia.dali.fn.expand_dims(sliced, axes=new_axes, new_axis_names=new_axis_names)

    def shape(self, *, dtype=None, device="cpu"):
        """Returns the run-time shapes of this DataNode as a new DataNode

        Parameters
        ----------
        arg_dtype : DALIDataType, optional
            If specified, the shape will be converted to this data type; defaults to INT64.
        device : str, optional
            The device ("cpu" or "gpu") where the result is returned; defaults to CPU.
        """
        from . import fn

        if device == "cpu":
            self._check_gpu2cpu()
        return fn._shape(self, dtype=dtype, device=device)

    def property(self, key, *, device="cpu"):
        """Returns a metadata property associated with a DataNode