如何在BERT中打印输出层的输出权重？

from transformers import BertModel, BertTokenizer

# 加载BERT模型和分词器
model_name = 'bert-base-uncased'
model = BertModel.from_pretrained(model_name)
tokenizer = BertTokenizer.from_pretrained(model_name)

text = "Hello, how are you?"
tokens = tokenizer.tokenize(text)
input_ids = tokenizer.convert_tokens_to_ids(tokens)

import torch

input_tensor = torch.tensor([input_ids])
outputs = model(input_tensor)

output = outputs[0]  # 获取模型的输出
output_weights = model.cls.predictions.decoder.weight  # 获取输出层的权重
print(output_weights)