log_device_placement參數

log_device_placement 它是Session的一個參數,設置該參數log_device_placement=True,讓咱們能夠看到咱們的tensor、op是在哪臺設備、哪顆CPU上運行的。python 示例代碼:web import tensorflow as tf import os os.environ['TF_CPP_MIN_LOG_LEVEL']='2' # 建立ten
相關文章
相關標籤/搜索