15 lines
345 B
Python
15 lines
345 B
Python
|
import warnings
|
||
|
|
|
||
|
|
|
||
|
|
def ignore_scalar_return_in_dp():
|
||
|
|
# Users get confused by this warning so we silence it
|
||
|
|
m_1 = """
|
||
|
|
Was asked to gather along dimension 0, but all
|
||
|
|
input tensors were scalars; will instead unsqueeze
|
||
|
|
and return a vector.
|
||
|
|
"""
|
||
|
|
warnings.filterwarnings('ignore', message=m_1)
|
||
|
|
|
||
|
|
|
||
|
|
ignore_scalar_return_in_dp()
|