print(x.size()[1]) # Print the column dimension. Output: 5 3. Note: You can use the basic operation of tuple to get the value of torch.size() ...
確定! 回上一頁