Transformers bert

>>> from transformers import BertTokenizer, BertModel
>>> import torch

>>> tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
>>> model = BertModel.from_pretrained('bert-base-uncased')

>>> inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
>>> outputs = model(**inputs)

>>> last_hidden_states = outputs.last_hidden_state

Similar codes

0 vote386 views1 answer
Shebang line python
Python
3 votes323 views2 answers
How to format a file in python
Python
3 votes344 views6 answers
Django login
Python
1 vote404 views2 answers
Operators in python
Python
15 votes345 views3 answers
Divide list into equal parts python
Python
59 votes353 views6 answers
How to split word in python
Python
103 votes338 views4 answers
Separate a string in python
Python
1 vote378 views1 answer
Using the Python string below to perform a search using a regular expression that you create. search_string=’’’This is a string to search for a regular expression like regular expression or regular-expression or regular:expression o
Python
3 votes289 views1 answer
Exp in python
Python
1 vote348 views1 answer
Python read integer from stdin
Python