输出tensor维度大小

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

    for text in atis_out:#text是二重矩阵
        atis_2 = text
        print(list(atis_2.size()))

你可能感兴趣的:(python编程)