/usr/bin/env/python
2 # coding=utf-8
3 import tensorflow as tf
4 import numpy as np
5
6 input_ids...={input_ids: [[1, 2], [3, 0], [8, 2], [5, 1]]}))
结果为:
[array([[0, 1, 2],
[3, 4, 5],
[6,...[15, 16, 17]]), array([[18, 19, 20],
[21, 22, 23],
[24, 25, 26]])]
[[[ 3 4 5]
[ 6...7 8]]
[[ 9 10 11]
[ 0 1 2]]
[[24 25 26]
[ 6 7 8]]
[[15 16 17]
[ 3 4 5]]]