输出tensor维度大小

	atis_out = bert_output(texts_atis, atis)#bert输出向量atis

    for text in atis_out:#text是二重矩阵
        atis_2 = text
        print(list(atis_2.size()))
发布了41 篇原创文章 · 获赞 44 · 访问量 7651

猜你喜欢

转载自blog.csdn.net/tailonh/article/details/105207694