bs=64 # batch size
xb = x_train[0:bs] # a mini-batch from x
preds = model(xb) # predictions
preds[0], preds.shape
yb = y_train[0:bs]
loss_func(preds, yb)
IndexError Traceback (последний последний вызов) в () ----> 1 loss_func (preds, yb)
в nll (вход, цель) 1 def nll (вход, target): ----> 2 return -input [range (target.shape [0]), target] .mean ()
IndexError: тензоры, используемые в качестве индексов, должны быть длинными, байтовыми или булевыми тензорами