Question about output pipeline(feature-extraction)
Hi,

I'm new to python and transformers so please bare with me. I have the following code to use the pipeline wrapper of transformers.

```
from transformers import (
    pipeline
)
import h5py

nlp = pipeline('feature-extraction', model='bert-base-cased', config='bert-base-cased', tokenizer='bert-base-cased', device=-1)

test = nlp("PersonA: Hi . PersonB: How are you doing ? PersonA: I 'm doing alright thank you very much.")

h5f = h5py.File('test.h5', 'a')
h5f.create_dataset('name', data=test)
h5f.close()
```

The above proof of concept works fine for me however I have two questions.

1. If I look at the shape of the h5py dataset it is (1, 270, 768). For my purpose I require the shape (270, 768). How can I make sure the output gets saved to h5 in this format?

2. device=-1 means the code will be executed on CPU correct?

Could someone please help me out with these two? ;)