Method flatten

warp/_src/types.py:4244–4267

Return a zero-copy view of the array collapsed to 1-D. Only supported for contiguous arrays.

(self)

            return warp.clone(self, device=device, requires_grad=requires_grad)

    def flatten(self):
        """Return a zero-copy view of the array collapsed to 1-D. Only supported for contiguous arrays."""
        if self.ndim == 1:
            return self

        if not self.is_contiguous:
            raise RuntimeError("Flattening non-contiguous arrays is unsupported.")

        a = array(
            ptr=self.ptr,
            dtype=self.dtype,
            shape=(self.size,),
            device=self.device,
            pinned=self.pinned,
            copy=False,
            grad=None if self.grad is None else self.grad.flatten(),
        )

        # transfer read flag
        a._is_read = self._is_read

        # store back-ref to stop data being destroyed
        a._ref = self
        return a

    def reshape(self, shape):
        """Return a reshaped array. Only supported for contiguous arrays.