【d2l】【常见函数】【3】 tensor.argmax( )

从张量的特定维度上取出最大的元素值所对应的索引
image

image

posted @ 2023-07-25 17:13  zz子木zz  阅读(43)  评论(0)    收藏  举报