@@ -0,0 +1,7 @@
import torch
import torch.nn as nn
import torch.nn.functional as nnf
import numpy as np
t = torch.tensor([1, 2, 3]).float()
print(torch.argmax(t))
The note is not visible to the blocked user.