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

lycnight / 2023-07-25 / 原文

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

image