rce > ( sample: FloatTensor  )   Parameters   sample (torch.FloatTensor of shape (batch_size, num_channels, sample_size)) —
The hidden states output from the last layer of the model.    The output of UNet1DModel.
