2

import numpy as np
import torch.nn
import tensorflow as tf


def main(labels):
    label = labels
    label_list = []
    for label in labels:
        # extend在label_list后面追加元素，map是对第二个参数的数据进行对于的操作变化
        label_list.extend(map(int, list(label)))
    label_list = np.array(label_list) + 1
    label_list = torch.Tensor(label_list)
    nammelist = [[2, 4,5], [1, 2, 3]]
    nammelist2 = ['4', '2']
    j = np.asarray(nammelist).max(1) #维度不一致回报错
    datas = tf.Variable(tf.random_uniform((3, 3), minval=0, maxval=1))
    with tf.Session() as session:
        session.run(tf.initialize_all_variables())
        print(datas.eval())
        datamax = tf.argmax(datas, axis=1)
        print(datamax.eval())
    datanp = np.arange(0,1.2,.2).reshape(2, 3)
    print(datanp)
    datanpmax = np.max(datanp, axis=1)
    print(datanpmax)
    print(datanp.max(1))

    print(nammelist == nammelist2)
    # print(label_list.type())
    # # 这个方法把一个list直接转换成一个tensor
    # print(torch.IntTensor([18] * int(32)))


if __name__ == '__main__':
    main([[1, 2, 3, 5, 2, 4]])