Size () [0] y_support_set one hot = Variable ( torch. Zeros (batch_size, sequence_length, classes per set). Scatter (2, y_support_set. data, 1), ...
確定! 回上一頁