Assuming you would like to check the device memory used to store the tensor data, you could multiply the tensor.nelement() with the number of bytes per element (4 for the default float32 dtype) and would get the actual size.